From 7ff2af6fe7bffea90dc4a31c93140c189917c659 Mon Sep 17 00:00:00 2001 From: Scott Date: Wed, 2 Feb 2022 13:47:36 -0800 Subject: Let's be explicit with asserts; there was a bug in histogram caused by assert foo when foo was an int with valid valid 0. --- ml/model_trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'ml/model_trainer.py') diff --git a/ml/model_trainer.py b/ml/model_trainer.py index 6fc0da0..a37885c 100644 --- a/ml/model_trainer.py +++ b/ml/model_trainer.py @@ -170,9 +170,9 @@ class TrainingBlueprint(ABC): print(msg) logger.info(msg) - assert best_training_score - assert best_test_score - assert best_params + assert best_training_score is not None + assert best_test_score is not None + assert best_params is not None ( scaler_filename, model_filename, -- cgit v1.3