Validation on loading model

tryolabs · Jun 28, 2024 · 29c5270 · 29c5270
1 parent 162dc22
commit 29c5270
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()