hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

remove TODO

Browse Source
This commit is contained in:
Alexander Soare
2024-03-21 13:51:26 +00:00
parent 98361073ef
commit 41912b962b
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/scripts/eval.py
View File

@@ -51,7 +51,6 @@ def eval_policy(
def maybe_render_frame(env: EnvBase, _): def maybe_render_frame(env: EnvBase, _):
if save_video or (return_first_video and i == 0): # noqa: B023 if save_video or (return_first_video and i == 0): # noqa: B023
# TODO now: generalize kwarg or maybe just remove it
ep_frames.append(env.render()) # noqa: B023 ep_frames.append(env.render()) # noqa: B023
with torch.inference_mode(): with torch.inference_mode():