From e7426a1946e2fdba9247250e9f5206fcacee8a40 Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Mon, 20 Jul 2020 18:14:09 +0200 Subject: [PATCH] Fix lost import --- stable_baselines3/td3/td3.py | 1 + 1 file changed, 1 insertion(+) diff --git a/stable_baselines3/td3/td3.py b/stable_baselines3/td3/td3.py index 530bc38..b0a5a21 100644 --- a/stable_baselines3/td3/td3.py +++ b/stable_baselines3/td3/td3.py @@ -12,6 +12,7 @@ from stable_baselines3.common.noise import ActionNoise from stable_baselines3.common.off_policy_algorithm import OffPolicyAlgorithm from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, RolloutReturn from stable_baselines3.common.utils import polyak_update, safe_mean +from stable_baselines3.common.vec_env import VecEnv from stable_baselines3.td3.policies import TD3Policy