From 4a804ced9dfd0a48b8d72a7f3094102d6ded0c3d Mon Sep 17 00:00:00 2001 From: Giacomo Spigler <spiglerg@gmail.com> Date: Fri, 26 Apr 2019 11:28:48 +0200 Subject: [PATCH] fixed set_transitions bug --- flatland/core/transition_map.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/flatland/core/transition_map.py b/flatland/core/transition_map.py index 78dd911..bd6e79a 100644 --- a/flatland/core/transition_map.py +++ b/flatland/core/transition_map.py @@ -181,7 +181,9 @@ class GridTransitionMap(TransitionMap): """ if len(cell_id) == 3: - self.transitions.set_transitions(self.grid[cell_id[0]][cell_id[1]], cell_id[2], new_transitions) + self.grid[cell_id[0]][cell_id[1]] = self.transitions.set_transitions(self.grid[cell_id[0]][cell_id[1]], + cell_id[2], + new_transitions) elif len(cell_id) == 2: self.grid[cell_id[0]][cell_id[1]] = new_transitions else: @@ -241,7 +243,10 @@ class GridTransitionMap(TransitionMap): print('GridTransitionMap.set_transition() ERROR: \ wrong cell_id tuple.') return - self.transitions.set_transition(self.grid[cell_id[0]][cell_id[1]], cell_id[2], transition_index, new_transition) + self.grid[cell_id[0]][cell_id[1]] = self.transitions.set_transition(self.grid[cell_id[0]][cell_id[1]], + cell_id[2], + transition_index, + new_transition) def save_transition_map(self, filename): """ -- GitLab