Update machine_learning_hep/processer.py

Co-authored-by: Jochen Klein <[email protected]>
alisw · Dec 21, 2023 · 62fa119 · 62fa119
1 parent fed5cee
commit 62fa119
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/machine_learning_hep/processer.py b/machine_learning_hep/processer.py
@@ -182,7 +182,7 @@ def __init__(self, case, datap, run_param, mcordata, p_maxfiles, # pylint: disab
         if os.path.isdir(self.d_root):
             self.l_path = list_folders(self.d_root, self.n_root, self.p_maxfiles,
                                        self.select_jobs)
-        elif is_file_indir(self.d_pkl, self.n_reco):
+        elif glob.glob(f"{self.d_pkl}/**/{self.n_reco}", recursive=True):
             self.l_path = list_folders(self.d_pkl, self.n_reco, self.p_maxfiles,
                                        self.select_jobs)
         else: