naming convention and running tests

2022-07-11 16:42:56 +02:00
parent ade83b5ae6
commit 123915e4fa
2 changed files with 14 additions and 296 deletions
@@ -71,8 +71,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
    def observation(self, observation):
        # return context space if we are
        mask = self.env.context_mask
-        if self.is_time_aware:
-            mask = np.append(mask, False)
+        # if self.is_time_aware:
+        #     mask = np.append(mask, False)
        obs = observation[mask] if self.return_context_observation else observation
        # cast dtype because metaworld returns incorrect that throws gym error
        return obs.astype(self.observation_space.dtype)