diff --git a/src/pystk2_gymnasium/envs.py b/src/pystk2_gymnasium/envs.py index f77d394..185ba24 100644 --- 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