ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 04:11:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup (#5)

Browse Source
This commit is contained in:
Jade Choghari
2025-08-28 22:49:32 +03:00
committed by GitHub
parent 440e22c184
commit cb18fc07ef
2 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lerobot/scripts/train.py
View File

@@ -301,8 +301,8 @@ def train(cfg: TrainPipelineConfig):
if eval_env:
# added by jade, close all env in multi eval setup
if cfg.env.multitask_eval:
for task_group, envs_dict in eval_env.items():
for idx, env in envs_dict.items():
for _task_group, envs_dict in eval_env.items():
for _idx, env in envs_dict.items():
env.close()
else:
eval_env.close()