saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-27 20:02:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove assert

Browse source
This commit is contained in:
Antonin RAFFIN 2020-08-05 12:18:16 +02:00
parent a5030f36b4
commit f2e129b829
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
stable_baselines3/common/buffers.py
View file

@ -420,7 +420,8 @@ class NstepReplayBuffer(ReplayBuffer):
self.gamma = gamma
def _get_samples(self, batch_inds: np.ndarray, env: Optional[VecNormalize] = None) -> ReplayBufferSamples:
assert not np.any(batch_inds == self.pos)
# TODO(PartiallyTyped): explain why this assert was here
# assert not np.any(batch_inds == self.pos)
actions = self.actions[batch_inds, 0, :]
gamma = self.gamma