Quality of life patches for eval.py (#86)

2026-05-17 17:50:09 +00:00 · 2024-04-19 12:33:47 +01:00
parent ac0ab27333
commit e2168163cd
2 changed files with 53 additions and 38 deletions
@@ -330,6 +330,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
                rollout_env,
                policy,
                transform=offline_dataset.transform,
+                return_episode_data=True,
                seed=cfg.seed,
            )