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

[lib.segmentation.scripts.predict] Fix test expectations

parent e5eea112
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
+ 1
− 0
View file @ 2ee2f060
...@@ -67,3 +67,4 @@ def predict( ...@@ -67,3 +67,4 @@ def predict(
json_predictions = run(model, datamodule, device_manager, output_folder) json_predictions = run(model, datamodule, device_manager, output_folder)
save_json_with_backup(output_folder / "predictions.json", json_predictions) save_json_with_backup(output_folder / "predictions.json", json_predictions)
logger.info(f"Predictions saved to `{str(json_predictions)}`")
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