diff --git a/stable_baselines3/td3/td3.py b/stable_baselines3/td3/td3.py index 95730fa..917c56f 100644 --- a/stable_baselines3/td3/td3.py +++ b/stable_baselines3/td3/td3.py @@ -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",