Skip to content

Commit

Permalink
bugfix ML method checker script
Browse files Browse the repository at this point in the history
  • Loading branch information
LonnekeScheffer committed May 24, 2024
1 parent 0426a2a commit 4eed865
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/check_new_ml_method.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,7 +212,7 @@ def check_store_load(ml_method_instance, tmp_path):
assert (orig_preds["my_label"] == loaded_preds["my_label"]).all(), f"When using the same example data to call .predict() on both the original model " \
f"and the model loaded after storing, different results were found: " \
f"\nOriginal model predictions: {orig_preds}" \
f"\nLoaded model predictions: {orig_preds}." \
f"\nLoaded model predictions: {loaded_preds}." \
f"\nThis indicates an issue with storing and loading the model."

logging.info(f"...The original and re-loaded model make the same predictions.")
Expand Down

0 comments on commit 4eed865

Please sign in to comment.