fix

SMLC-NYSBC · May 8, 2024 · 84feab3 · 84feab3
1 parent 517a919
commit 84feab3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tardis_em/utils/predictor.py b/tardis_em/utils/predictor.py
@@ -1128,7 +1128,7 @@ def __call__(self, *args, **kwargs):
             self.log_tardis(id_, i, log_id=3)
 
             self.preprocess_DIST(id_name=i)
-            self.segments = None
+            self.segments = np.zeros((0, 4))
 
             # Break iter loop for instances if no point cloud is found
             if len(self.pc_ld) == 0: