Fix possible division by zero

GAA-UAM · Apr 13, 2024 · 4bd8713 · 4bd8713
1 parent e91419d
commit 4bd8713
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/skfda/misc/scoring.py b/skfda/misc/scoring.py
@@ -152,7 +152,12 @@ def _integral_average_fdatairregular(
 
     integrals = np.mean(score.integrate(), axis=1)
     lebesgue_measures = np.diff(score.sample_range, axis=-1).reshape(-1)
-    normalized_integrals = integrals / lebesgue_measures
+    normalized_integrals = np.divide(
+        integrals,
+        lebesgue_measures,
+        out=np.zeros_like(integrals),
+        where=lebesgue_measures != 0,
+    )
     return np.average(normalized_integrals, weights=weights)