bugfix when muultiple images are predicted

conda-build/meta-v0.2.5.yaml

@@ -0,0 +1,52 @@
+package:
+    name: "tardis-em"
+    version: "0.2.5"
+
+source:
+#    git_url: https://github.com/SMLC-NYSBC/TARDIS.git
+#    git_tag: v0.2.5
+    path: ../
+
+build:
+    number: 0
+#    noarch: python
+    platform: osx-arm64
+    script: {{ PYTHON }} -m pip install . -vv
+
+requirements:
+    host:
+        - pip
+        - python>=3.10,<3.12
+    build:
+        - python>=3.10,<3.12
+        - pip
+        - setuptools
+    run:
+        - python>=3.10, <3.12
+        - pytorch>=1.12.0
+        - numpy>=1.21.0
+        - pandas>1.3.0
+        - tifffile>2021.11.0
+        - imagecodecs
+        - scikit-learn>1.0.1
+        - scikit-image>0.19.2
+        - scipy>=1.8.1
+        - pillow
+        - open3d
+        - requests>2.28.0
+        - ipython>8.0.0
+        - click>8.0.4
+        - plyfile
+        - pyvista
+
+extra:
+  recipe-maintainers:
+    - RRobert92
+    - tbepler
+
+about:
+    doc_url: https://smlc-nysbc.github.io/TARDIS/
+    dev_url: https://github.com/SMLC-NYSBC/TARDIS
+    home: https://github.com/SMLC-NYSBC/TARDIS
+    license: MIT
+    license_file: LICENSE

conda-build/meta.yaml

@@ -1,6 +1,6 @@
 package:
     name: "tardis-em"
-    version: "0.2.5"
+    version: "0.2.6"
 
 source:
 #    git_url: https://github.com/SMLC-NYSBC/TARDIS.git
@@ -9,8 +9,8 @@ source:
 
 build:
     number: 0
-#    noarch: python
-    platform: osx-arm64
+    noarch: python
+#    platform: osx-arm64
     script: {{ PYTHON }} -m pip install . -vv
 
 requirements:

docs/HISTORY.md

@@ -1,11 +1,12 @@
 # History
 <details open>
-    <summary><b>0.2.5 (2024-05-22)</b></summary>
+    <summary><b>0.2.6 (2024-05-22)</b></summary>
 
 * **General changes:**
     * *General*:
-        * Added support for predicting Actin
-        * Predicting point clouds directly from cli
+      * Bugfix when multiple files are predicted in batch
+      * Added support for predicting Actin
+      * Predicting point clouds directly from cli
 
 </details>
 

tardis_em/_version.py

@@ -8,4 +8,4 @@
 #  MIT License 2021 - 2024                                            #
 # #####################################################################
 
-version = "0.2.5"
+version = "0.2.6"

tardis_em/utils/predictor.py

@@ -712,7 +712,11 @@ def log_tardis(self, id_: int, i: Union[str, np.ndarray], log_id: float):
             "text_3": f"Image {id_ + 1}/{len(self.predict_list)}: {i}",
         }
 
-        no_segments = np.max(self.segments[:, 0]) if len(self.segments) > 0 else 0
+        if log_id == 7:
+            no_segments = np.max(self.segments[:, 0]) if len(self.segments) > 0 else 0
+        else:
+            no_segments = 'None'
+
         # Define text configurations for each log_id
         text_configurations = {
             0: {