import gym
from tianshou.env import DummyVectorEnv

if __name__ == "__main__":
    def create_env():
        env = gym.make("CartPole-v1")
        print(type(env))  # <class 'gym.wrappers.time_limit.TimeLimit'>
        return env

    train_envs = DummyVectorEnv(
        [create_env]
    )

    train_envs.set_env_attr("test_attribute", 1337)
    print(train_envs.get_env_attr("test_attribute"))  # Returns: [1337]
    print(hasattr(train_envs.workers[0].env, "test_attribute"))  # Returns: True
    print(hasattr(train_envs.workers[0].env.unwrapped, "test_attribute"))  # Returns: False