Let's be explicit with asserts; there was a bug in histogram

[python_utils.git] / ml / model_trainer.py

diff --git a/ml/model_trainer.py b/ml/model_trainer.py
index 6fc0da071e7dd6e2e026b8bbb7d7780c81015985..a37885ce3e92f20d11bb61f4268407689829bca4 100644 (file)
--- 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,