Skip to content

Commit 9fac235

Browse files
authored
Update common.py to fix test errors
1 parent 766a475 commit 9fac235

File tree

1 file changed

+0
-5
lines changed
  • src/imitation/algorithms/adversarial

1 file changed

+0
-5
lines changed

src/imitation/algorithms/adversarial/common.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -507,11 +507,6 @@ def train_gen_with_disc(
507507
)
508508
self._global_step += 1
509509

510-
gen_trajs, ep_lens = self.venv_buffering.pop_trajectories()
511-
self._check_fixed_horizon(ep_lens)
512-
gen_samples = rollout.flatten_trajectories_with_rew(gen_trajs)
513-
self._gen_replay_buffer.store(gen_samples)
514-
515510
def train(
516511
self,
517512
total_timesteps: int,

0 commit comments

Comments
 (0)