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

Maintainer and pos_dicts fixed. Are sets now.

Browse Source
This commit is contained in:
Steffen Illium
2023-11-10 14:16:48 +01:00
parent 6711a0976b
commit 9b289591ba
22 changed files with 100 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
marl_factory_grid/modules/destinations/rules.py
View File

@ -106,7 +106,7 @@ class SpawnDestinationsPerAgent(Rule):
super(Rule, self).__init__()
self.per_agent_positions = {key: [ast.literal_eval(x) for x in val] for key, val in coords_or_quantity.items()}
def on_init(self, state, lvl_map):
def on_reset(self, state, lvl_map):
for (agent_name, position_list) in self.per_agent_positions.items():
agent = h.get_first(state[c.AGENT], lambda x: agent_name in x.name)
assert agent