From 7a702c3cf27bd74e6ca7f6390f0a95dcccc24b42 Mon Sep 17 00:00:00 2001 From: romue <robert.mueller@campus.lmu.de> Date: Tue, 18 May 2021 11:32:49 +0200 Subject: [PATCH] added scaling to simple_factory_getting_dirty.py rendering --- environments/factory/simple_factory_getting_dirty.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/environments/factory/simple_factory_getting_dirty.py b/environments/factory/simple_factory_getting_dirty.py index 93421bd..f14e79a 100644 --- a/environments/factory/simple_factory_getting_dirty.py +++ b/environments/factory/simple_factory_getting_dirty.py @@ -39,7 +39,8 @@ class GettingDirty(BaseFactory): height, width = self.state.shape[1:] self.renderer = Renderer(width, height, view_radius=0) - dirt = [Entity('dirt', [x, y], self.state[DIRT_INDEX, x, y]) for x, y in np.argwhere(self.state[DIRT_INDEX] > h.IS_FREE_CELL)] + 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)]