Replaced x.size by x.shape to make it work for n-dimensional arrays

ecmwf-projects · Dec 18, 2024 · 25976e9 · 25976e9
1 parent 927e9d6
commit 25976e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ibicus/utils/_math_utils.py b/ibicus/utils/_math_utils.py
@@ -279,7 +279,7 @@ def cdf(self, x: np.ndarray, *fit: tuple) -> np.ndarray:
         else:
             return np.where(
                 x == 0,
-                np.random.uniform(0, p0, x.size),
+                np.random.uniform(0, p0, x.shape),
                 p0 + (1 - p0) * self.distribution.cdf(x, *fit_rainy_days),
             )
 
@@ -423,7 +423,7 @@ def cdf(self, x: np.ndarray, *fit: tuple) -> np.ndarray:
         """
         x = np.where(
             x < self.censoring_threshold,
-            np.random.uniform(0, self.censoring_threshold, x.size),
+            np.random.uniform(0, self.censoring_threshold, x.shape),
             x,
         )
         return scipy.stats.gamma.cdf(x, *fit)