Make attribute getting behaviour of DummyEnvWorker compatible with gymnasium v1

Author: opcode81

test/base/test_env.py

@@ -213,8 +213,6 @@ def test_attr_unwrapped() -> None:
     train_envs = DummyVectorEnv([lambda: gym.make("CartPole-v1")])
     train_envs.set_env_attr("test_attribute", 1337)
     assert train_envs.get_env_attr("test_attribute") == [1337]
-    # mypy doesn't know but BaseVectorEnv takes the reserved keys in gym.Env (one of which is env)
-    assert hasattr(train_envs.workers[0].env, "test_attribute")  # type: ignore
     assert hasattr(train_envs.workers[0].env.unwrapped, "test_attribute")  # type: ignore
 
 

tianshou/env/worker/dummy.py

@@ -15,7 +15,7 @@ def __init__(self, env_fn: Callable[[], gym.Env]) -> None:
         super().__init__(env_fn)
 
     def get_env_attr(self, key: str) -> Any:
-        return getattr(self.env, key)
+        return getattr(self.env.unwrapped, key)
 
     def set_env_attr(self, key: str, value: Any) -> None:
         setattr(self.env.unwrapped, key, value)