Strange bug! Add {"new_step_api": True} in gym environments.

2023-07-05 15:39:22 +08:00
parent 7bf46575f4
commit 2cc72bb188
5 changed files with 18 additions and 3 deletions
--- a/evox_adaptor/gym_no_distribution.py
+++ b/evox_adaptor/gym_no_distribution.py
@@ -55,9 +55,7 @@ class Gym(Problem):
        return -fitnesses, State(key=key)

    def __rollout(self, seeds, pop):
-        observations, infos = zip(
-            *[env.reset(seed=seed) for env, seed in zip(self.envs, seeds)]
-        )
+        observations = [env.reset(seed=seed) for env, seed in zip(self.envs, seeds)]
        terminates, truncates = np.zeros((2, self.pop_size), dtype=bool)
        fitnesses, rewards = np.zeros((2, self.pop_size))