From f2e129b829f99ea63420363bb6d60af024b8f7d8 Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Wed, 5 Aug 2020 12:18:16 +0200 Subject: [PATCH] Remove assert --- stable_baselines3/common/buffers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stable_baselines3/common/buffers.py b/stable_baselines3/common/buffers.py index e69da39..ad940bb 100644 --- a/stable_baselines3/common/buffers.py +++ b/stable_baselines3/common/buffers.py @@ -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