From f1bba678e154c38792bcfef5a79f6489bba85a0f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Quentin=20Gallou=C3=A9dec?= <45557362+qgallouedec@users.noreply.github.com> Date: Tue, 29 Nov 2022 17:18:34 +0100 Subject: [PATCH] Fix `sync_envs_normalization` --- stable_baselines3/common/vec_env/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/stable_baselines3/common/vec_env/__init__.py b/stable_baselines3/common/vec_env/__init__.py index be4326e..6e704d9 100644 --- a/stable_baselines3/common/vec_env/__init__.py +++ b/stable_baselines3/common/vec_env/__init__.py @@ -67,7 +67,9 @@ def sync_envs_normalization(env: "GymEnv", eval_env: "GymEnv") -> None: """ env_tmp, eval_env_tmp = env, eval_env while isinstance(env_tmp, VecEnvWrapper): + assert isinstance(eval_env_tmp, VecEnvWrapper) # Ensure that env and eval_env are similarly wrapped. if isinstance(env_tmp, VecNormalize): + assert isinstance(eval_env_tmp, VecNormalize) # Ensure that env and eval_env are similarly wrapped. # Only synchronize if observation normalization exists if hasattr(env_tmp, "obs_rms"): eval_env_tmp.obs_rms = deepcopy(env_tmp.obs_rms)