mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-14 10:48:45 +00:00
17 lines
482 B
Python
17 lines
482 B
Python
|
import gym
|
||
|
|
|
||
|
|
|
||
|
|
class OrderEnforcing(gym.Wrapper):
|
||
|
|
def __init__(self, env):
|
||
|
|
super(OrderEnforcing, self).__init__(env)
|
||
|
|
self._has_reset = False
|
||
|
|
|
||
|
|
def step(self, action):
|
||
|
|
assert self._has_reset, "Cannot call env.step() before calling reset()"
|
||
|
|
observation, reward, done, info = self.env.step(action)
|
||
|
|
return observation, reward, done, info
|
||
|
|
|
||
|
|
def reset(self, **kwargs):
|
||
|
|
self._has_reset = True
|
||
|
|
return self.env.reset(**kwargs)
|