fix: ZeroDivisionError when predicting on small sample file (#68)

instadeepai · Jan 21, 2025 · dca4423 · dca4423
1 parent d2b926b
commit dca4423
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/instanovo/transformer/predict.py b/instanovo/transformer/predict.py
@@ -256,9 +256,10 @@ def get_preds(
     delta = time.time() - start
 
     logger.info(f"Time taken for {data_path} is {delta:.1f} seconds")
-    logger.info(
-        f"Average time per batch (bs={config['batch_size']}): {delta/len(dl):.1f} seconds"
-    )
+    if len(dl) > 0:
+        logger.info(
+            f"Average time per batch (bs={config['batch_size']}): {delta/len(dl):.1f} seconds"
+        )
 
     if not denovo:
         pred_df["targets"] = targs