Explicitly track total agents in model data collection

Princeton-CDH · Feb 6, 2024 · 693da8d · 693da8d
1 parent 6e130c7
commit 693da8d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/simulatingrisk/hawkdove/model.py b/simulatingrisk/hawkdove/model.py
@@ -259,6 +259,8 @@ def get_data_collector_options(self):
                 "percent_hawk": "percent_hawk",
                 "rolling_percent_hawk": "rolling_percent_hawk",
                 "status": "status",
+                # explicitly track total agents, instead of inferring from grid size
+                "total_agents": "num_agents",
             },
             "agent_reporters": {
                 "risk_level": "risk_level",