Merge pull request #47 from jjshoots/fix_observation_space

jjshoots · web-flow · commit e62f9b62a059 · 2024-06-28T14:41:26.000+09:00
Fix observation space for flatten waypoint wrapper
diff --git a/PyFlyt/gym_envs/utils/flatten_waypoint_env.py b/PyFlyt/gym_envs/utils/flatten_waypoint_env.py
@@ -32,7 +32,9 @@ def __init__(self, env: Env, context_length=2):
             0
         ]  # type: ignore [reportGeneralTypeIssues]
         self.observation_space = Box(
-            low=-np.inf, high=np.inf, shape=(self.attitude_shape + self.target_shape,)
+            low=-np.inf,
+            high=np.inf,
+            shape=(self.attitude_shape + self.target_shape * self.context_length,),
         )
 
     def observation(self, observation) -> np.ndarray:
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "PyFlyt"
-version = "0.22.0"
+version = "0.22.1"
 authors = [
   { name="Jet", email="taijunjet@hotmail.com" },
 ]

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,9 @@ def __init__(self, env: Env, context_length=2):`
`32`	`32`	`0`
`33`	`33`	`] # type: ignore [reportGeneralTypeIssues]`
`34`	`34`	`self.observation_space = Box(`
`35`		`- low=-np.inf, high=np.inf, shape=(self.attitude_shape + self.target_shape,)`
	`35`	`+ low=-np.inf,`
	`36`	`+ high=np.inf,`
	`37`	`+ shape=(self.attitude_shape + self.target_shape * self.context_length,),`
`36`	`38`	`)`
`37`	`39`
`38`	`40`	`def observation(self, observation) -> np.ndarray:`
Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"`
`4`	`4`
`5`	`5`	`[project]`
`6`	`6`	`name = "PyFlyt"`
`7`		`-version = "0.22.0"`
	`7`	`+version = "0.22.1"`
`8`	`8`	`authors = [`
`9`	`9`	`{ name="Jet", email="[email protected]" },`
`10`	`10`	`]`