Skip to content
Snippets Groups Projects
Commit 9ae6fc92 authored by Andri Joos's avatar Andri Joos :blush:
Browse files

fix env observation shape

parent 0608cd70
No related merge requests found
......@@ -13,7 +13,7 @@ from tf_agents.specs.array_spec import BoundedArraySpec, ArraySpec
class HelloWorldEnv(PyEnvironment):
def __init__(self, ip: str, handle_auto_reset: bool = False):
self._action_spec = BoundedArraySpec(shape=(), dtype=np.int32, minimum=0, maximum=2, name='action')
self._observation_spec = ArraySpec(shape=(1,), dtype=np.float32, name='observation')
self._observation_spec = ArraySpec(shape=(1,), dtype=np.float64, name='observation')
self._client = airsim.MultirotorClient(ip=ip)
super(HelloWorldEnv, self).__init__(handle_auto_reset)
......@@ -53,7 +53,7 @@ class HelloWorldEnv(PyEnvironment):
def _getObservation(self):
geo_point = self._client.getGpsData().gnss.geo_point
# return geo_point.latitude, geo_point.longitude, geo_point.altitude
return (geo_point.altitude,)
return np.array([geo_point.altitude])
@staticmethod
def _calculate_reward(obs: tuple[float, float, float]):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment