From 2db4479eb66b6ba3f5557ff51bb07f0c1621bf1d Mon Sep 17 00:00:00 2001
From: Robert Kiewisz <56911280+RRobert92@users.noreply.github.com>
Date: Sun, 2 Jul 2023 14:03:22 +0200
Subject: [PATCH] fix back compatibilit for DIST

---
 tardis_pytorch/utils/predictor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tardis_pytorch/utils/predictor.py b/tardis_pytorch/utils/predictor.py
index d62e663a..1c471c09 100644
--- 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(