mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-13 02:17:14 +00:00
21 lines
465 B
Python
21 lines
465 B
Python
|
import gym
|
||
|
|
from gym.vector import make
|
||
|
|
from gym.vector import VectorEnvWrapper
|
||
|
|
|
||
|
|
class DummyWrapper(VectorEnvWrapper):
|
||
|
|
def __init__(self, env):
|
||
|
|
self.env = env
|
||
|
|
self.counter = 0
|
||
|
|
|
||
|
|
def reset_async(self):
|
||
|
|
super().reset_async()
|
||
|
|
self.counter += 1
|
||
|
|
|
||
|
|
|
||
|
|
def test_vector_env_wrapper_inheritance():
|
||
|
|
env = make('FrozenLake-v0', asynchronous=False)
|
||
|
|
wrapped = DummyWrapper(env)
|
||
|
|
wrapped.reset()
|
||
|
|
assert wrapped.counter == 1
|
||
|
|
|