diff --git a/cooperative_cuisine/state_representation.py b/cooperative_cuisine/state_representation.py index 0a0aeef53d8c68af8c7642d8225c31d97ff196f0..cbc70c51642cfa9994cbcbffe6cd00f50e5f36d2 100644 --- a/cooperative_cuisine/state_representation.py +++ b/cooperative_cuisine/state_representation.py @@ -211,7 +211,7 @@ def create_movement_graph(state: StateRepresentation, diagonal=True) -> Graph: free_space = np.ones((width, height), dtype=bool) for counter in state["counters"]: grid_idx = np.array(counter["pos"]).round().astype(int) - free_space[*grid_idx] = False + free_space[grid_idx[0], grid_idx[1]] = False graph = networkx.Graph() for i in range(width):