use ordered dict to make compatible with python 3.5

mesaroom · Aug 23, 2020 · 48f7e12 · 48f7e12
1 parent f530b41
commit 48f7e12
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mesa/batchrunner.py b/mesa/batchrunner.py
@@ -12,6 +12,7 @@
 from multiprocessing import Pool, cpu_count
 import pandas as pd
 from tqdm import tqdm
+from collections import OrderedDict
 
 
 class ParameterError(TypeError):
@@ -109,8 +110,9 @@ def __init__(
         if self.agent_reporters:
             self.agent_vars = {}
 
-        self.datacollector_model_reporters = {}
-        self.datacollector_agent_reporters = {}
+        #Make Compatible with Python 3.5
+        self.datacollector_model_reporters = OrderedDict()
+        self.datacollector_agent_reporters = OrderedDict()
 
         self.display_progress = display_progress
 

diff --git a/tests/test_batchrunner.py b/tests/test_batchrunner.py
@@ -140,7 +140,6 @@ def test_agent_level_vars(self):
         agent_collector = batch.get_collector_agents()
         # extra columns with run index and agentId
         expected_cols = (len(self.variable_params) + len(self.agent_reporters) + 2)
-
         self.assertEqual(agent_vars.shape, (self.model_runs * NUM_AGENTS, expected_cols))
 
         self.assertEqual(