From 809b6740107fbecb06a50d32857fefd1bc1b7867 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Sun, 23 May 2021 11:40:01 +0200 Subject: [PATCH] Pytype fixes --- stable_baselines3/common/off_policy_algorithm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_baselines3/common/off_policy_algorithm.py b/stable_baselines3/common/off_policy_algorithm.py index e92ad71..1e526d1 100644 --- a/stable_baselines3/common/off_policy_algorithm.py +++ b/stable_baselines3/common/off_policy_algorithm.py @@ -509,7 +509,7 @@ class OffPolicyAlgorithm(BaseAlgorithm): for i in range(len(dones)): if isinstance(next_obs, dict): - replay_buffer.add( + replay_buffer.add( # pytype: disable=wrong-arg-types {key: obs[i] for key, obs in self._last_original_obs.items()}, {key: next_obs_[i] for key, next_obs_ in next_obs.items()}, buffer_action[i],