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

rht · web-flow · commit 4bf4b528883f · 2024-02-22T06:48:33.000-05:00
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
 

Original file line number	Diff line number	Diff line change
`@@ -184,7 +184,9 @@ def get_reports(agent):`
`184`	`184`
`185`	`185`	`agent_records = map(`
`186`	`186`	`get_reports,`
`187`		`- model.schedule.agents if hasattr(model, "schedule") else model.agents,`
	`187`	`+ model.schedule.agents`
	`188`	`+ if hasattr(model, "schedule") and model.schedule is not None`
	`189`	`+ else model.agents,`
`188`	`190`	`)`
`189`	`191`	`return agent_records`
`190`	`192`