Update machine_learning_hep/processer.py

Co-authored-by: Jochen Klein <[email protected]>
alisw · Feb 7, 2025 · 54b15a8 · 54b15a8
1 parent b185ab0
commit 54b15a8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/machine_learning_hep/processer.py b/machine_learning_hep/processer.py
@@ -248,10 +248,9 @@ def __init__(self, case, datap, run_param, mcordata, p_maxfiles, # pylint: disab
                             for ipt in range(self.p_nptbins)]
         self.f_evt_count_ml = os.path.join(self.d_pkl_ml, self.n_evt_count_ml)
 
-        if self.n_gen_sl:
-            self.lpt_gensk_sl = [self.n_gen_sl.replace(".p", "_%s%d_%d.p" % \
-                          (self.v_var_binning, self.lpt_anbinmin[i], self.lpt_anbinmax[i])) \
-                          for i in range(self.p_nptbins)]
+          self.lpt_gensk_sl = [self.n_gen_sl.replace(".p", "_%s%d_%d.p" %
+                        (self.v_var_binning, self.lpt_anbinmin[i], self.lpt_anbinmax[i]))
+                        for i in range(self.p_nptbins)] if self.n_gen_sl else None
 
         self.lpt_recodec = None
         if self.doml is True: