Check if label is explicitly None because it can be also 0

cta-observatory · Apr 15, 2021 · d10dac3 · d10dac3
1 parent 95cff80
commit d10dac3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/protopipe/mva/utils.py b/protopipe/mva/utils.py
@@ -44,7 +44,7 @@ def prepare_data(ds, derived_features, cuts, select_data=True, label=None):
     # This is always useful
     ds["log10_true_energy"] = np.log10(ds["true_energy"])
 
-    if label:  # only for classification
+    if label is not None:  # only for classification
         ds["label"] = np.full(len(ds), label)
 
         # This is needed because our reference analysis uses energy as