Skip to content

Commit f4ce58c

Browse files
Add message to test-model
1 parent 4d86b56 commit f4ce58c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

bioimageio/core/__main__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,10 @@ def test_model(
5757
if len(devices) == 0:
5858
devices = None
5959
test_passed = prediction.test_model(model_rdf, weight_format=weight_format, devices=devices, decimal=decimal)
60+
if test_passed:
61+
print(f"Model test for {model_rdf} has passed.")
62+
else:
63+
print(f"Model test for {model_rdf} has FAILED!")
6064
ret_code = 0 if test_passed else 1
6165
return ret_code
6266

0 commit comments

Comments
 (0)