Fix add_cost bug

Returned matrix was completely wrong if we ended up fixing costs so they are above 1
eladyaniv01 · Feb 3, 2021 · bc5052e · bc5052e
1 parent 776ca67
commit bc5052e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MapAnalyzer/Pather.py b/MapAnalyzer/Pather.py
@@ -370,9 +370,9 @@ def add_cost(position: Tuple[float, float], radius: float, arr: ndarray, weight:
             arr[disk] = np.where(arr[disk] == 1, initial_default_weights, arr[disk])
 
         arr[disk] += weight
-        if safe and np.any(arr < 1):
+        if safe and np.any(arr[disk] < 1):
             logger.warning(
                     "You are attempting to set weights that are below 1. falling back to the minimum (1)")
-            arr = np.where(arr[disk] < 1, 1, arr[disk])
+            arr[disk] = np.where(arr[disk] < 1, 1, arr[disk])
 
         return arr