add useful print statements for batch processing

talmolab · Apr 11, 2024 · 2f43876 · 2f43876
1 parent 6941e71
commit 2f43876
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sleap_roots/trait_pipelines.py b/sleap_roots/trait_pipelines.py
@@ -647,6 +647,7 @@ def compute_batch_traits(
         """
         all_traits = []
         for plant in plants:
+            print(f"Processing series: {plant.series_name}")
             # Compute frame level traits for the plant.
             plant_traits = self.compute_plant_traits(plant)
 
@@ -664,6 +665,7 @@ def compute_batch_traits(
 
         if write_csv:
             all_traits.to_csv(csv_path, index=False)
+            print(f"Batch traits saved to {csv_path}")
         return all_traits
 
     def compute_batch_multiple_dicots_traits(