only try to start env if it is closed

2024-03-22 15:32:55 +00:00
parent e698d38a35
commit d43fa600a0
1 changed files with 2 additions and 1 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -90,6 +90,7 @@ def eval_policy(
        if policy is not None:
            policy.clear_action_queue()

+        if env.is_closed:
            env.start()  # needed to be able to get the seeds the first time as BatchedEnvs are lazy
        seeds.extend(env._next_seed)
        with torch.inference_mode():