debug benchmark

trackmania-rl · Mar 30, 2023 · 8d9689c · 8d9689c
1 parent 7fc88de
commit 8d9689c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tmrl/networking.py b/tmrl/networking.py
@@ -714,7 +714,7 @@ def run_env_benchmark(self, nb_steps, test=False):
         for _ in range(nb_steps):
             obs, rew, terminated, truncated, info = self.step(obs=obs, test=test, collect_samples=False)
             if terminated or truncated:
-                obs, info = self.reset(collect_samples=False)
+                break
         print_with_timestamp(f"Benchmark results:\n{self.env.benchmarks()}")
 
     def send_and_clear_buffer(self):