Updated lines 393-403 with: def Dispersion(self): """Calculate the dispersion of the swarm""" # Ensure self.gpos is a NumPy array if not isinstance(self.gpos, np.ndarray): self.gpos = np.array(self.gpos) # Now self.gpos should be a NumPy array, so we can calculate the dispersion x, y = self.gpos[:, 0], self.gpos[:, 1] dx = x.max() - x.min() dy = y.max() - y.min() return (dx + dy) / 2.0

app.py

@@ -385,10 +385,22 @@ class GWO:
         #dy = y.max() - y.min()
         #return (dx + dy) / 2.0
 
-    def Dispersion(self):
+    #def Dispersion(self):
         # """Calculate the dispersion of the swarm"""
-        dispersion = np.std(self.gpos[:, 0]) + np.std(self.gpos[:, 1])
-        return dispersion
+        #dispersion = np.std(self.gpos[:, 0]) + np.std(self.gpos[:, 1])
+        #return dispersion
+
+    def Dispersion(self):
+        """Calculate the dispersion of the swarm"""
+        # Ensure self.gpos is a NumPy array
+        if not isinstance(self.gpos, np.ndarray):
+            self.gpos = np.array(self.gpos)
+        
+        # Now self.gpos should be a NumPy array, so we can calculate the dispersion
+        x, y = self.gpos[:, 0], self.gpos[:, 1]
+        dx = x.max() - x.min()
+        dy = y.max() - y.min()
+        return (dx + dy) / 2.0
 
     def plot_dispersion(self):
         """Plot the dispersion over time"""