Updated lines 343-346 with: # Ensure wolf_positions is a 2D array wolf_positions = np.array(wolf_positions) if wolf_positions.ndim == 1: wolf_positions = wolf_positions.reshape(-1, 1)

app.py

@@ -340,6 +340,11 @@ class GWO:
             "vpos": self.vpos
             }
     def plot_contour_and_wolves(self, wolf_positions):
+         # Ensure wolf_positions is a 2D array
+        wolf_positions = np.array(wolf_positions)
+        if wolf_positions.ndim == 1:
+            wolf_positions = wolf_positions.reshape(-1, 1)
+            
         # Define the objective function
         def objective_function(x, y):
             return -5.0*np.exp(-0.5*((x+2.2)**2/0.4+(y-4.3)**2/0.4)) + -2.0*np.exp(-0.5*((x-2.2)**2/0.4+(y+4.3)**2/0.4))