fix: include total batch size in inference completion message

abhishek-pattern · abhishek-pattern · commit 5b91333156c2 · 2026-03-02T14:09:25.000+05:30
diff --git a/src/ds_platform_utils/metaflow/batch_inference_pipeline.py b/src/ds_platform_utils/metaflow/batch_inference_pipeline.py
@@ -234,7 +234,7 @@ def inference_worker():
                 with _timer(f"🔹 Generated predictions for file {file_id}"):
                     predictions_df = predict_fn(df)
                 inference_queue.put((file_id, predictions_df), timeout=timeout_per_batch)
-                print(f"🔘 Inference completed for batch {file_id}")
+                print(f"🔘 Inference completed for batch {file_id} of {len(file_batches)}")
 
         def upload_worker():
             while True:
