Fix do_syst_ml condition

alisw · Feb 21, 2025 · 26ee38c · 26ee38c
1 parent 239ad72
commit 26ee38c
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/machine_learning_hep/steer_analysis.py b/machine_learning_hep/steer_analysis.py
@@ -293,11 +293,16 @@ 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: