saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-02 23:40:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Pytype fixes

Browse source
This commit is contained in:
Antonin Raffin 2021-05-23 11:40:01 +02:00
parent 43b4ee9cd7
commit 809b674010
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
stable_baselines3/common/off_policy_algorithm.py
View file

@ -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],