From 84feab3580c45a7e124a84c3d3638af3e90b0626 Mon Sep 17 00:00:00 2001
From: Robert Kiewisz <56911280+RRobert92@users.noreply.github.com>
Date: Wed, 8 May 2024 12:51:08 +0200
Subject: [PATCH] fix

---
 tardis_em/utils/predictor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tardis_em/utils/predictor.py b/tardis_em/utils/predictor.py
index 8f318d81..a98078f0 100644
--- 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: