From 65cf5c5c6f4916dd1a489bc6d5a17ab21b5ca20c Mon Sep 17 00:00:00 2001 From: romue Date: Tue, 11 May 2021 17:45:27 +0200 Subject: [PATCH] updated base factory --- environments/factory/base_factory.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/environments/factory/base_factory.py b/environments/factory/base_factory.py index 3bd1286..a34ad9f 100644 --- a/environments/factory/base_factory.py +++ b/environments/factory/base_factory.py @@ -63,9 +63,7 @@ class BaseFactory: pos_x, pos_y = pos collisions_vec = self.state[:, pos_x, pos_y].copy() # "vertical fiber" at position of agent i collisions_vec[h.AGENT_START_IDX + agent_i] = h.IS_FREE_CELL # no self-collisions - if valid: - pass - else: + if not valid: collisions_vec[h.LEVEL_IDX] = h.IS_OCCUPIED_CELL return collisions_vec @@ -82,9 +80,7 @@ class BaseFactory: # Does not collide width level boundrys self.move(agent_i, old_pos, new_pos) return new_pos, valid - else: - # Agent seems to be trying to collide in this step - return old_pos, valid + return old_pos, valid @property def free_cells(self) -> np.ndarray: