@@ -812,12 +812,6 @@ if(ROOT_pyroot_FOUND)
812812 # Disable tutorial showing connection to the HTCondor service at CERN
813813 list (APPEND pyveto analysis/dataframe/distrdf004_dask_lxbatch.py)
814814
815- ROOT_FIND_PYTHON_MODULE(xgboost QUIET )
816- if (NOT ROOT_XGBOOST_FOUND OR NOT dataframe)
817- file (GLOB tmva_veto_py RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} machine_learning/tmva10*.py roofit/roofit/rf618_mixture_models.py)
818- list (APPEND pyveto ${tmva_veto_py} )
819- endif ()
820-
821815 if (NOT tmva-pymva)
822816 file (GLOB tmva_veto_py RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} machine_learning/keras/*.py)
823817 list (APPEND pyveto ${tmva_veto_py} )
@@ -923,10 +917,15 @@ if(ROOT_pyroot_FOUND)
923917 machine_learning/pytorch/*.py
924918 machine_learning/RBatchGenerator_PyTorch.py
925919 )
926- file (GLOB requires_xgboost RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} machine_learning/tmva10*.py)
920+ file (GLOB requires_xgboost RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
921+ machine_learning/tmva101_Training.py
922+ roofit/roofit/rf618_mixture_models.py
923+ )
927924 file (GLOB requires_sklearn RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
928925 machine_learning/TMVA_SOFIE_Models.py
926+ machine_learning/tmva101_Training.py # uses the xgboost sklearn plugin
929927 roofit/roofit/rf617_simulation_based_inference_multidimensional.py
928+ roofit/roofit/rf618_mixture_models.py # uses the xgboost sklearn plugin
930929 )
931930 file (GLOB requires_tensorflow RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
932931 machine_learning/RBatchGenerator_TensorFlow.py
0 commit comments