steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-20 03:08:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

added agent assets

Browse Source
This commit is contained in:
romue
2021-05-18 12:16:31 +02:00
parent a7fcf249cb
commit 31ee7f8a6d
7 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
environments/factory/simple_factory_getting_dirty.py
View File

@ -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)