Small adjustment and a not was missing rules

2025-06-23 20:11:34 +02:00 · 2023-10-25 17:01:58 +02:00
parent 9420986230
commit 85d65a2c78
3 changed files with 3 additions and 2 deletions
--- a/marl_factory_grid/modules/batteries/rules.py
+++ b/marl_factory_grid/modules/batteries/rules.py
@ -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]: