Add NaN check

bcgov · Jul 23, 2024 · d27bca2 · d27bca2
1 parent 2f9126e
commit d27bca2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/app/weather_models/machine_learning.py b/api/app/weather_models/machine_learning.py
@@ -6,6 +6,7 @@
 from typing import List
 from logging import getLogger
 from sklearn.linear_model import LinearRegression
+import math
 import numpy as np
 from sqlalchemy.orm import Session
 from app.weather_models import SCALAR_MODEL_VALUE_KEYS
@@ -263,6 +264,9 @@ def predict_precipitation(self, model_precipitation: float, timestamp: datetime)
         if model_precipitation is None:
             logger.warning('model precipitation for %s was None', timestamp)
             return None
+        if math.isnan(model_precipitation):
+            logger.warning("model precipitation for %s was NaN", timestamp)
+            return None
         hour = timestamp.hour
         predicted_precip_24h = self.regression_models_v2._precip_model.predict(hour, [[model_precipitation]])
         if predicted_precip_24h is None or len(predicted_precip_24h) == 0: