From 273ddc75035cb45a7df2cc85d9efb68b9797034d Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 26 Jun 2024 13:21:59 +0200
Subject: [PATCH] [libs.segmentation.scripts.predict] Fix printout
---
src/mednet/libs/segmentation/scripts/predict.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/mednet/libs/segmentation/scripts/predict.py b/src/mednet/libs/segmentation/scripts/predict.py
index db43e0e1..b15d7daf 100644
--- a/src/mednet/libs/segmentation/scripts/predict.py
+++ b/src/mednet/libs/segmentation/scripts/predict.py
@@ -66,5 +66,6 @@ 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)}`")
+ predictions_file = output_folder / "predictions.json"
+ save_json_with_backup(predictions_file, json_predictions)
+ logger.info(f"Predictions saved to `{str(predictions_file)}`")
--
GitLab