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.

1 files changed, 3 insertions, 3 deletions

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,