From 311e136d678ade59fe722bf1880e25383136c9ce Mon Sep 17 00:00:00 2001 From: Andri Joos <andri@joos.io> Date: Thu, 28 Sep 2023 20:12:53 +0200 Subject: [PATCH] env return exact type for specs --- HelloWorldEnv.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/HelloWorldEnv.py b/HelloWorldEnv.py index fc09bbf..154e135 100644 --- a/HelloWorldEnv.py +++ b/HelloWorldEnv.py @@ -17,10 +17,10 @@ class HelloWorldEnv(PyEnvironment): self._client = airsim.MultirotorClient(ip=ip) super(HelloWorldEnv, self).__init__(handle_auto_reset) - def action_spec(self) -> NestedArraySpec: + def action_spec(self) -> BoundedArraySpec: return self._action_spec - def observation_spec(self) -> NestedArraySpec: + def observation_spec(self) -> ArraySpec: return self._observation_spec def _reset(self) -> TimeStep: -- GitLab