Minor fix

This commit is contained in:
Antonin RAFFIN 2020-07-16 18:16:10 +02:00
parent e379a7a4c8
commit 5db5e4738c

View file

@ -308,8 +308,8 @@ class TD3(OffPolicyAlgorithm):
return self
def collect_rollouts(
self, # noqa: C901
def collect_rollouts( # noqa: C901
self,
env: VecEnv,
# Type hint as string to avoid circular import
callback: "BaseCallback",