diff --git a/tutorials/CMakeLists.txt b/tutorials/CMakeLists.txt index 965c0247fd136..2a91b9f917b42 100644 --- a/tutorials/CMakeLists.txt +++ b/tutorials/CMakeLists.txt @@ -304,6 +304,7 @@ else() find_python_module(keras QUIET) find_python_module(sonnet QUIET) find_python_module(graph_nets QUIET) + find_python_module(sklearn QUIET) if (NOT PY_KERAS_FOUND) list(APPEND tmva_veto tmva/TMVA_SOFIE_Keras.C) list(APPEND tmva_veto tmva/TMVA_SOFIE_Keras_HiggsModel.C) @@ -315,6 +316,9 @@ else() list(APPEND tmva_veto tmva/TMVA_SOFIE_RSofieReader.C) list(APPEND tmva_veto tmva/RBatchGenerator_TensorFlow.py) endif() + if (NOT PY_SKLEARN_FOUND) + list(APPEND tmva_veto tmva/TMVA_SOFIE_Models.py) + endif() if (NOT PY_TORCH_FOUND) list(APPEND tmva_veto tmva/TMVA_SOFIE_PyTorch.C) list(APPEND tmva_veto tmva/RBatchGenerator_PyTorch.py)