Fix do_syst_ml condition

alisw · Feb 20, 2025 · f8e2772 · f8e2772
1 parent 11d4ece
commit f8e2772
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/machine_learning_hep/steer_analysis.py b/machine_learning_hep/steer_analysis.py
@@ -290,14 +290,15 @@ def mlhepmod(name):
 
     analyzers = ana_mgr.get_analyzers()
     # For ML WP systematics
-    if mltype == "MultiClassification":
-        syst_ml_pt_cl0 = syst_class(data_param[case], case, typean, analyzers,
-                                    mymultiprocessmc, mymultiprocessdata, 0)
-        syst_ml_pt_cl1 = syst_class(data_param[case], case, typean, analyzers,
-                                    mymultiprocessmc, mymultiprocessdata, 1)
-    else:
-        syst_ml_pt = syst_class(data_param[case], case, typean, analyzers,
-                                mymultiprocessmc, mymultiprocessdata)
+    if do_syst_ml:
+        if mltype == "MultiClassification":
+            syst_ml_pt_cl0 = syst_class(data_param[case], case, typean, analyzers,
+                                        mymultiprocessmc, mymultiprocessdata, 0)
+            syst_ml_pt_cl1 = syst_class(data_param[case], case, typean, analyzers,
+                                        mymultiprocessmc, mymultiprocessdata, 1)
+        else:
+            syst_ml_pt = syst_class(data_param[case], case, typean, analyzers,
+                                    mymultiprocessmc, mymultiprocessdata)
 
     #perform the analysis flow
     if dodownloadalice: