remove TODO

This commit is contained in:
Alexander Soare
2024-03-21 13:51:26 +00:00
parent 98361073ef
commit 41912b962b

View File

@@ -51,7 +51,6 @@ def eval_policy(
def maybe_render_frame(env: EnvBase, _):
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
with torch.inference_mode():