[tmva][sofie] Add dependency on sklearn for TMVA_SOFIE_Models tutorial

roofit-dev · Sep 6, 2023 · 879c228 · 879c228
1 parent 5e38958
commit 879c228
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)