Merge pull request #10 from tryolabs/hotfix/extra-model-validation-on…

…-predict Validation on loading model
tryolabs · Jun 28, 2024 · c87507a · c87507a
2 parents 162dc22 + 29c5270
commit c87507a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/challenge/model.py b/challenge/model.py
@@ -150,5 +150,7 @@ def predict(
         Returns:
             (List[int]): predicted targets.
         """
+        if self._model is None:
+            self.__load_model(MODEL_PATH)
         predictions = self._model.predict(features)
         return predictions.tolist()