Skip to content
Snippets Groups Projects
Commit 273ddc75 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[libs.segmentation.scripts.predict] Fix printout

parent 15b88354
No related branches found
No related tags found
1 merge request!46Create common library
Hide whitespace changes
Inline Side-by-side
src/mednet/libs/segmentation/scripts/predict.py
+ 3
− 2
View file @ 273ddc75
......@@ -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)}`")
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment