add divide by zero check

antonky · antonky · commit 7ce005570518 · 2021-03-02T20:51:29.000-05:00
diff --git a/sklearn/gaussian_process/_gpr.py b/sklearn/gaussian_process/_gpr.py
@@ -200,7 +200,8 @@ def fit(self, X, y):
             self._y_train_std = np.std(y, axis=0)
 
             # Remove mean and make unit variance
-            y = (y - self._y_train_mean) / self._y_train_std
+            # avoid division by 0 if standard deviation is 0
+            y = (y - self._y_train_mean) / (self._y_train_std if self._y_train_std != 0 else 1)
 
         else:
             self._y_train_mean = np.zeros(1)
