diff --git a/environments/factory/assets/agent1.png b/environments/factory/assets/agent1.png index ce41a32..0574988 100644 Binary files a/environments/factory/assets/agent1.png and b/environments/factory/assets/agent1.png differ diff --git a/environments/factory/assets/agent2.png b/environments/factory/assets/agent2.png index fe232e9..af03c8c 100644 Binary files a/environments/factory/assets/agent2.png and b/environments/factory/assets/agent2.png differ diff --git a/environments/factory/assets/agent3.png b/environments/factory/assets/agent3.png index 60cf5ee..e83b474 100644 Binary files a/environments/factory/assets/agent3.png and b/environments/factory/assets/agent3.png differ diff --git a/environments/factory/assets/agent4.png b/environments/factory/assets/agent4.png index e3a928f..092560d 100644 Binary files a/environments/factory/assets/agent4.png and b/environments/factory/assets/agent4.png differ diff --git a/environments/factory/assets/agent5.png b/environments/factory/assets/agent5.png index 0630ecd..d565aec 100644 Binary files a/environments/factory/assets/agent5.png and b/environments/factory/assets/agent5.png differ diff --git a/environments/factory/assets/agent6.png b/environments/factory/assets/agent6.png index 05e2645..36484dd 100644 Binary files a/environments/factory/assets/agent6.png and b/environments/factory/assets/agent6.png differ diff --git a/environments/factory/simple_factory_getting_dirty.py b/environments/factory/simple_factory_getting_dirty.py index 7a6d948..e966475 100644 --- a/environments/factory/simple_factory_getting_dirty.py +++ b/environments/factory/simple_factory_getting_dirty.py @@ -42,9 +42,9 @@ class GettingDirty(BaseFactory): dirt = [Entity('dirt', [x, y], min(self.state[DIRT_INDEX, x, y], 1), 'scale') for x, y in np.argwhere(self.state[DIRT_INDEX] > h.IS_FREE_CELL)] walls = [Entity('dirt', pos) for pos in np.argwhere(self.state[h.LEVEL_IDX] > h.IS_FREE_CELL)] - agents = [Entity('agent', pos) for pos in np.argwhere(self.state[h.AGENT_START_IDX] > h.IS_FREE_CELL)] + agents = [Entity('agent1', pos) for pos in np.argwhere(self.state[h.AGENT_START_IDX] > h.IS_FREE_CELL)] - self.renderer.render(OrderedDict(dirt=dirt, wall=walls, agent=agents)) + self.renderer.render(OrderedDict(dirt=dirt, wall=walls, agent1=agents)) def spawn_dirt(self) -> None: free_for_dirt = self.free_cells(excluded_slices=DIRT_INDEX)