MAINT: Initialize distance for robustness.

pmgbergen · Nov 5, 2023 · 284b9c6 · 284b9c6
1 parent c7c646d
commit 284b9c6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/darsia/measure/wasserstein.py b/src/darsia/measure/wasserstein.py
@@ -1212,6 +1212,9 @@ def _solve(self, flat_mass_diff: np.ndarray) -> tuple[float, np.ndarray, dict]:
             self.darcy_init.copy(), rhs.copy(), solution_i
         )
 
+        # Initialize distance in case below iteration fails
+        new_distance = 0
+
         # Initialize container for storing the convergence history
         convergence_history = {
             "distance": [],
@@ -1489,6 +1492,9 @@ def _solve(self, flat_mass_diff: np.ndarray) -> tuple[float, np.ndarray, dict]:
             self.darcy_init.copy(), rhs.copy(), solution_i
         )
 
+        # Initialize distance in case below iteration fails
+        new_distance = 0
+
         # Initialize container for storing the convergence history
         convergence_history = {
             "distance": [],