fix back compatibilit for DIST

SMLC-NYSBC · Jul 2, 2023 · 2db4479 · 2db4479
1 parent 0863943
commit 2db4479
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tardis_pytorch/utils/predictor.py b/tardis_pytorch/utils/predictor.py
@@ -74,8 +74,6 @@ def __init__(
         self,
         predict: str,
         dir_: str,
-        checkpoint: list,
-        feature_size: Optional[float],
         output_format: str,
         patch_size: int,
         cnn_threshold: float,
@@ -91,6 +89,8 @@ def __init__(
         instances: bool,
         device_: str,
         debug: bool,
+        checkpoint: Optional[list] = None,
+        feature_size: Optional[float] = None,
     ):
         if predict not in ["Filament", "Membrane2D", "Membrane", "Microtubule"]:
             TardisError(