Skip to content

Commit

Permalink
and more typos
Browse files Browse the repository at this point in the history
  • Loading branch information
bpiwowar committed Oct 27, 2023
1 parent 9d43308 commit 8825c58
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/pystk2_gymnasium/envs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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):
Expand All @@ -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()}
Expand Down

0 comments on commit 8825c58

Please sign in to comment.