From c9bc695c117ee273e4280fd8713020a39d7f5c53 Mon Sep 17 00:00:00 2001 From: madsmpedersen <m@madsp.dk> Date: Fri, 3 Jun 2016 11:28:49 +0200 Subject: [PATCH] small changes in high_level.py --- wetb/dlc/high_level.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/wetb/dlc/high_level.py b/wetb/dlc/high_level.py index a67327e..98ac733 100644 --- a/wetb/dlc/high_level.py +++ b/wetb/dlc/high_level.py @@ -180,11 +180,14 @@ class DLCHighLevel(object): return fatigue_dist def files_dict(self): + fatigue_dlcs = self.dlc_df[['F' in str(l).upper() for l in self.dlc_df['load']]]['dlc'] + if len(fatigue_dlcs) == 0: + return {} if not hasattr(self, "res_folder") or self.res_folder == "": files = glob.glob(os.path.join(self.res_path, "*.sel")) + glob.glob(os.path.join(self.res_path, "*/*.sel")) else: files = [] - fatigue_dlcs = self.dlc_df[['F' in str(l).upper() for l in self.dlc_df['load']]]['dlc'] + for dlc_id in fatigue_dlcs: dlc_id = str(dlc_id) if "%" in self.res_folder: -- GitLab