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

maintainer test

Browse Source
This commit is contained in:
Chanumask
2023-11-16 19:45:25 +01:00
parent f25f90a78b
commit 26318a7b48
4 changed files with 84 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -9,7 +9,6 @@ from marl_factory_grid.environment import constants as c
from marl_factory_grid.modules.destinations import constants as d
from marl_factory_grid.modules.destinations.entitites import Destination
from marl_factory_grid.utils.states import Gamestate
ANY = 'any'
@ -118,7 +117,7 @@ class SpawnDestinationsPerAgent(Rule):
per_agent_d = {agent_name: [ast.literal_eval(x) for x in value]}
self.per_agent_positions.update(**per_agent_d)
def on_reset(self, state: Gamestate):
def on_reset(self, state):
for (agent_name, coords_or_quantity) in self.per_agent_positions.items():
agent = h.get_first(state[c.AGENT], lambda x: agent_name in x.name)
assert agent