bp 2fs seems to work

2022-05-31 19:41:08 +02:00
parent 2ea7f6a2ed
commit 59b15e82ea
5 changed files with 98 additions and 15 deletions
@@ -223,6 +223,9 @@ class EpisodicWrapper(gym.Env, ABC):
    def get_observation_from_step(self, observation: np.ndarray) -> np.ndarray:
        return observation[self.active_obs]

+    def seed(self, seed=None):
+        self.env.seed(seed)
+
    def plot_trajs(self, des_trajs, des_vels):
        import matplotlib.pyplot as plt
        import matplotlib