diff options
Diffstat (limited to 'ml')
| -rw-r--r-- | ml/model_trainer.py | 6 |
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, |