Caused by a duplicate range() call in experiments/irl_trainer.py.
On line 71:
self._random_range = range(expert_obs.shape[0])
Then on line 150:
indices = random.sample(range(self._random_range), self._irl.batch_size)
Removing this duplicate call resolved the issue for me. As far as I can tell this looks like a bug, as line 150 is the only reference to _random_range.
Caused by a duplicate range() call in experiments/irl_trainer.py.
On line 71:
self._random_range = range(expert_obs.shape[0])
Then on line 150:
indices = random.sample(range(self._random_range), self._irl.batch_size)
Removing this duplicate call resolved the issue for me. As far as I can tell this looks like a bug, as line 150 is the only reference to _random_range.