Merge pull request #205 from kywch/float-obs

Changed obs dtype to float32 (from float64) to match the obs space definition
MyoHub · Aug 16, 2024 · 3fe5c31 · 3fe5c31
2 parents 3a245b1 + 9f53244
commit 3fe5c31
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/myosuite/envs/obs_vec_dict.py b/myosuite/envs/obs_vec_dict.py
@@ -78,10 +78,10 @@ def obsdict2obsvec(self, obs_dict, ordered_obs_keys):
         if not self.initialized:
             self.initialize(obs_dict, ordered_obs_keys)
 
-        # recover vec
-        obsvec = np.zeros(0)
+        obs_list = [np.zeros(0)]
         for key in self.ordered_obs_keys:
-            obsvec = np.concatenate([obsvec, obs_dict[key].ravel()]) # ravel helps with images
+            obs_list.append(obs_dict[key].ravel())  # ravel helps with images
+        obsvec = np.concatenate(obs_list, dtype=np.float32)
 
         # cache
         t = obs_dict['time']