make new_agents optional for recreatertree

projectmesa · May 27, 2020 · 64d0bbb · 64d0bbb
1 parent 8f8468d
commit 64d0bbb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mesa_geo/geospace.py b/mesa_geo/geospace.py
@@ -142,8 +142,11 @@ def get_neighbors(self, agent):
         neighbors = [self.agents[i] for i in neighbors_idx]
         return neighbors
 
-    def _recreate_rtree(self, new_agents):
+    def _recreate_rtree(self, new_agents=None):
         """Create a new rtree index from agents shapes."""
+
+        if new_agents is None:
+            new_agents = []
         old_agents = list(self.agents)
         agents = old_agents + new_agents