Fix undefined variable

HumanCompatibleAI · Oct 7, 2023 · 7f04e79 · 7f04e79
1 parent 407b7dd
commit 7f04e79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/imitation/scripts/ingredients/demonstrations.py b/src/imitation/scripts/ingredients/demonstrations.py
@@ -146,7 +146,7 @@ def _generate_expert_trajs(
     with environment.make_rollout_venv() as env:  # type: ignore[wrong-arg-count]
         return rollout.rollout(
             expert.get_expert_policy(env),
-            rollout_env,
+            env,
             rollout.make_sample_until(min_episodes=n_expert_demos),
             rng=_rnd,
         )