Actually handle Zero division

force return value again
SALib · Sep 9, 2020 · da99fda · da99fda
1 parent e7128ec
commit da99fda
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/SALib/analyze/radial_ee.py b/src/SALib/analyze/radial_ee.py
@@ -80,7 +80,10 @@ def analyze(problem: Dict, X: np.array, Y: np.array, sample_sets: int,
         x_tmp = (X_base[:, i] - X[pos::nth, i])
 
         # Convert zero division nan to 0
-        ee[:, i] = np.nan_to_num((Y_base - Y[pos::nth]) / x_tmp)
+        try:
+            ee[:, i] = np.nan_to_num((Y_base - Y[pos::nth]) / x_tmp)
+        except ZeroDivisionError:
+            ee[:, i] = 0.0
     # End for
 
     Si = ResultDict((k, [None] * num_vars)