1
0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2026-04-24 07:47:32 +02:00

Small adjustment and a not was missing rules

This commit is contained in:
Steffen Illium
2023-10-25 17:01:58 +02:00
parent 9420986230
commit 85d65a2c78
3 changed files with 3 additions and 2 deletions
@@ -14,6 +14,7 @@ class Btry(Rule):
self.initial_charge = initial_charge
def on_init(self, state, lvl_map):
assert len(state[c.AGENT]), "There are no agents, did you already spawn them?"
state[b.BATTERIES].spawn(state[c.AGENT], self.initial_charge)
def tick_pre_step(self, state) -> List[TickResult]: