Replace defaultdict with OrderedDict

jokeradac · Dec 16, 2017 · 17ed18b · 17ed18b
1 parent c18aa50
commit 17ed18b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/examples/sugarscape_cg/sugarscape/schedule.py b/examples/sugarscape_cg/sugarscape/schedule.py
@@ -1,5 +1,5 @@
 import random
-from collections import defaultdict
+from collections import OrderedDict
 
 from mesa.time import RandomActivation
 
@@ -14,11 +14,11 @@ class RandomActivationByBreed(RandomActivation):
 
     Assumes that all agents have a step() method.
     '''
-    agents_by_breed = defaultdict(list)
+    agents_by_breed = OrderedDict(dict)
 
     def __init__(self, model):
         super().__init__(model)
-        self.agents_by_breed = defaultdict(list)
+        self.agents_by_breed = OrderedDict(dict)
 
     def add(self, agent):
         '''

diff --git a/examples/wolf_sheep/wolf_sheep/schedule.py b/examples/wolf_sheep/wolf_sheep/schedule.py
@@ -1,5 +1,5 @@
 import random
-from collections import defaultdict
+from collections import OrderedDict
 
 from mesa.time import RandomActivation
 
@@ -14,11 +14,11 @@ class RandomActivationByBreed(RandomActivation):
 
     Assumes that all agents have a step() method.
     '''
-    agents_by_breed = defaultdict(dict)
+    agents_by_breed = OrderedDict(dict)
 
     def __init__(self, model):
         super().__init__(model)
-        self.agents_by_breed = defaultdict(dict)
+        self.agents_by_breed = OrderedDict(dict)
 
     def add(self, agent):
         '''