diff --git a/src/pystk2_gymnasium/envs.py b/src/pystk2_gymnasium/envs.py index fae53b4..36b2a2e 100644 --- a/src/pystk2_gymnasium/envs.py +++ b/src/pystk2_gymnasium/envs.py @@ -244,7 +244,7 @@ def close(self): def step( self, action: STKAction - ) -> tuple[pystk2.WorldState, float, bool, bool, Dict[str, Any]]: + ) -> Tuple[pystk2.WorldState, float, bool, bool, Dict[str, Any]]: if self.use_ai: self.race.step() else: @@ -513,7 +513,7 @@ def to_discrete(self, action): def step( self, action: STKDiscreteAction - ) -> tuple[Any, float, bool, bool, Dict[str, Any]]: + ) -> Tuple[Any, float, bool, bool, Dict[str, Any]]: return super().step(self.from_discrete(action)) class OnlyContinuousActionSTKRaceEnv(SimpleSTKRaceEnv): @@ -535,7 +535,7 @@ def __init__(self, acceleration_steps=5, steer_steps=5, **kwargs): def step( self, action: Dict - ) -> tuple[Any, float, bool, bool, Dict[str, Any]]: + ) -> Tuple[Any, float, bool, bool, Dict[str, Any]]: return super().step({ **action, **{key: 0 for key, value in self.discrete_actions.items()}