added dtype in observation spaces

2021-12-01 15:55:38 +01:00
parent ca90f257d4
commit f9e7a34eda
4 changed files with 7 additions and 7 deletions
@@ -111,7 +111,7 @@ class ViaPointReacherEnv(BaseReacherDirectEnv):
            self.end_effector - self._via_point,
            self.end_effector - self._goal,
            self._steps
-        ])
+        ]).astype(np.float32)

    def _check_collisions(self) -> bool:
        return self._check_self_collision()