Assert to protect

[ci skip]
bpiwowar · Nov 16, 2023 · 5f8ed1d · 5f8ed1d
1 parent 9d3f3bd
commit 5f8ed1d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pystk2_gymnasium/envs.py b/src/pystk2_gymnasium/envs.py
@@ -473,6 +473,7 @@ def __init__(self, *, agents: List[AgentSpec] = None, **kwargs):
 
         # Setup the variables
         self.agents = agents
+        assert len(self.agents) <= self.num_kart, f"Too many agents ({len(self.agents)}) for {self.num_kart} karts"
 
         # Kart index for each agent (set when the race is setup)
         self.kart_indices = None