Merge pull request #33 from MPI-Dortmund/labeling_bug

fix labeling bug
MPI-Dortmund · Jul 17, 2023 · 1f12b50 · 1f12b50
2 parents cb36f72 + 7ad1390
commit 1f12b50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tomotwin/modules/common/preprocess.py b/tomotwin/modules/common/preprocess.py
@@ -385,7 +385,7 @@ def label_filename(path: str) -> str:
     filename = os.path.basename(path)
     try:
         lbl = re.search("id(?P<PDB>\d[a-zA-Z0-9]{3})", filename).group("PDB").upper()
-    except AssertionError:
+    except AttributeError:
         lbl = re.search("\d[a-zA-Z0-9]{3}", filename).group(0).upper()  # https://regex101.com/r/rZi0TZ/1
 
     return lbl