minor fixes in text output

bwmr · Sep 26, 2024 · 091dc6c · 091dc6c
1 parent 4b0a307
commit 091dc6c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/sizepicker/interface.py b/sizepicker/interface.py
@@ -2,7 +2,7 @@
 
 import click
 
-from .picker import Picker
+from sizepicker.picker import Picker
 
 
 @click.command()
@@ -108,7 +108,7 @@ def cli(radius,
         tomo = Path(tomo)
         output_dir = Path(output_dir)
 
-        print(f"working on {tomo.name}")
+        print(f"Working on {tomo.name}")
 
         p = Picker(tomo,
                    output_dir,
@@ -122,14 +122,14 @@ def cli(radius,
                          minsize_cont,
                          dilate_cont)
 
-        print(f"Contamination Mask written to {output_dir.stem}")
+        print(f"Contamination mask written to {output_dir.stem}")
 
         boxes, particles = p.detect(mask,
                                     radius_times,
                                     inhibit,
                                     detection_z)
 
-        print(f"Found {len(boxes)} Initial Picks")
+        print(f"Found {len(boxes)} initial picks.")
 
         metrics, threshold = p.prefilt(particles,
                                        stdtimes_pick)

diff --git a/sizepicker/picker.py b/sizepicker/picker.py
@@ -412,6 +412,8 @@ def filt(self,boxes,particles_metrics,stdthreshold,remove_edge):
             boxs_ZYX+=b
 
         # Transform from ZYX to XYZ
+        # TODO: handle case where no particles are left after filtering
+
         boxs_ZYX =  np.stack(boxs_ZYX)
         boxs_XYZ = boxs_ZYX[:,[2,1,0]]