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

small error fixes reset sanity and self sanity almost lost

Browse Source
This commit is contained in:
Steffen Illium
2023-11-10 17:40:27 +01:00
parent 9b289591ba
commit b6ab6ab652
7 changed files with 30 additions and 31 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_reset(self, state, lvl_map):
def on_reset(self, state):
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