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