fix: Use model.schedule only when it is not None (projectmesa#2050)

quaquel · Apr 9, 2024 · 368d3f7 · 368d3f7
1 parent f6211ae
commit 368d3f7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mesa/datacollection.py b/mesa/datacollection.py
@@ -184,7 +184,9 @@ def get_reports(agent):
 
         agent_records = map(
             get_reports,
-            model.schedule.agents if hasattr(model, "schedule") else model.agents,
+            model.schedule.agents
+            if hasattr(model, "schedule") and model.schedule is not None
+            else model.agents,
         )
         return agent_records