feat: close environment and log completion message in benchmark run

ll7 · Nov 27, 2024 · 46870d8 · 46870d8
1 parent c2f6685
commit 46870d8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/benchmark02.py b/scripts/benchmark02.py
@@ -110,6 +110,9 @@ def run_standardized_benchmark(
     avg_step_time = np.mean(step_times)
     steps_per_sec = 1.0 / avg_step_time
 
+    env.close()
+    logger.info("Benchmark run complete. env closed. return metrics")
+
     # System info
     system_info = {
         "platform": platform.platform(),