From 2ee2f0607b6cde85cf6e6cdcd5129ab952329836 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 26 Jun 2024 11:05:53 +0200
Subject: [PATCH] [lib.segmentation.scripts.predict] Fix test expectations
---
src/mednet/libs/segmentation/scripts/predict.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/mednet/libs/segmentation/scripts/predict.py b/src/mednet/libs/segmentation/scripts/predict.py
index 84c50d47..db43e0e1 100644
--- a/src/mednet/libs/segmentation/scripts/predict.py
+++ b/src/mednet/libs/segmentation/scripts/predict.py
@@ -67,3 +67,4 @@ def predict(
json_predictions = run(model, datamodule, device_manager, output_folder)
save_json_with_backup(output_folder / "predictions.json", json_predictions)
+ logger.info(f"Predictions saved to `{str(json_predictions)}`")
--
GitLab