fix back compatibilit for DIST

SMLC-NYSBC · Jul 2, 2023 · 79398fd · 79398fd
1 parent 97a0073
commit 79398fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tardis_pytorch/utils/predictor.py b/tardis_pytorch/utils/predictor.py
@@ -167,7 +167,7 @@ def __init__(
             sys.exit()
         # Searching for available images for prediction
         available_format = (".tif", ".mrc", ".rec", ".am")
-        omit_format = ("mask.tif", "mask.mrc", "mask.rec", 'Correlation_Lines.am')
+        omit_format = ("mask.tif", "mask.mrc", "mask.rec", 'Correlation_Lines.am', 'mask.am')
         self.output = join(self.dir, "temp", "Predictions")
         self.am_output = join(self.dir, "Predictions")