File size: 367 Bytes
b05c680 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import gym
from typing import Any
class IsVectorEnv(gym.Wrapper):
"""
Override to set properties to match gym.vector.VectorEnv
"""
def __init__(self, env: Any) -> None:
super().__init__(env)
self.is_vector_env = True
self.single_observation_space = env.observation_space
self.single_action_space = env.action_space
|