improve iter_cell_list_contents

rambam613 · Feb 5, 2021 · cccd438 · cccd438
1 parent 2a917aa
commit cccd438
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mesa/space.py b/mesa/space.py
@@ -310,7 +310,10 @@ def iter_cell_list_contents(
             An iterator of the contents of the cells identified in cell_list
 
         """
-        return (self[x][y] for x, y in cell_list if not self.is_cell_empty((x, y)))
+        for (x, y) in cell_list:
+            content = self.grid[x][y]
+            if not content == self.default_val():
+                yield content
 
     @accept_tuple_argument
     def get_cell_list_contents(