1 env

src/evaluation.py

@@ -347,7 +347,7 @@ def evaluate(model_id, revision):
 
     # Evaluate the agent on the environments
     for env_id in env_ids:
-        envs = gym.vector.SyncVectorEnv([make(env_id) for _ in range(3)])
+        envs = gym.vector.SyncVectorEnv([make(env_id) for _ in range(1)])
         observations, _ = envs.reset()
         episodic_returns = []
         while len(episodic_returns) < NUM_EPISODES: