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
@@ -106,7 +106,7 @@ class HoleReacherEnv(BaseReacherDirectEnv):
            # self._tmp_hole_depth,
            self.end_effector - self._goal,
            self._steps
-        ])
+        ]).astype(np.float32)

    def _get_line_points(self, num_points_per_link=1):
        theta = self._joint_angles[:, None]