Skip to content

Commit 8effe7e

Browse files
authored
Update model.py
1 parent fe6bc82 commit 8effe7e

File tree

1 file changed

+2
-3
lines changed
  • examples/boltzmann_wealth_model_network/boltzmann_wealth_model_network

1 file changed

+2
-3
lines changed

examples/boltzmann_wealth_model_network/boltzmann_wealth_model_network/model.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(self, unique_id, model):
5757
def move(self):
5858
possible_steps = [
5959
node
60-
for node in self.model.grid.get_neighbors(self.pos, include_center=False)
60+
for node in self.model.grid.get_neighborhood(self.pos, include_center=False)
6161
if self.model.grid.is_cell_empty(node)
6262
]
6363
if len(possible_steps) > 0:
@@ -66,8 +66,7 @@ def move(self):
6666

6767
def give_money(self):
6868

69-
neighbors_nodes = self.model.grid.get_neighbors(self.pos, include_center=False)
70-
neighbors = self.model.grid.get_cell_list_contents(neighbors_nodes)
69+
neighbors = self.model.grid.get_neighbors(self.pos, include_center=False)
7170
if len(neighbors) > 0:
7271
other = self.random.choice(neighbors)
7372
other.wealth += 1

0 commit comments

Comments
 (0)