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

DoneOnReach Rule added.

Browse Source
This commit is contained in:
Steffen Illium
2023-09-01 13:59:45 +02:00
parent 714e07a816
commit 178cc70ed2
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,6 +53,20 @@ class DestinationDone(Rule):
return [] return []
class DoneOnReach(Rule):
def __init__(self):
super(DoneOnReach, self).__init__()
def on_check_done(self, state) -> List[DoneResult]:
dests = [x.pos for x in state[d.DESTINATION]]
agents = [x.pos for x in state[c.AGENT]]
if any([x in dests for x in agents]):
return [DoneResult(self.name, validity=c.VALID, reward=r.DEST_REACHED)]
return [DoneResult(self.name, validity=c.NOT_VALID, reward=0)]
class DestinationSpawn(Rule): class DestinationSpawn(Rule):
def __init__(self, spawn_frequency: int = 5, n_dests: int = 1, def __init__(self, spawn_frequency: int = 5, n_dests: int = 1,