File size: 322 Bytes
f0ac49c
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from typing import Any

from rl_algo_impls.wrappers.vectorable_wrapper import VecotarableWrapper


class IsVectorEnv(VecotarableWrapper):
    """
    Override to set properties to match gym.vector.VectorEnv
    """

    def __init__(self, env: Any) -> None:
        super().__init__(env)
        self.is_vector_env = True