From dfb5b44dc9a694e7e493fb81898020ba0c388558 Mon Sep 17 00:00:00 2001 From: kywch Date: Mon, 11 Mar 2024 22:33:44 -0700 Subject: [PATCH] fixed error --- pokemonred_puffer/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py index d7945c0..47f4095 100644 --- a/pokemonred_puffer/train.py +++ b/pokemonred_puffer/train.py @@ -70,7 +70,7 @@ def env_creator( reward_config: pufferlib.namespace, ) -> pufferlib.emulation.GymnasiumPufferEnv: env = reward_class(env_config, reward_config) - for cfg, (_, wrapper_class) in zip(wrappers_config, wrapper_classes): + for cfg, (_, wrapper_class) in zip(wrappers_config, wrapper_classes.items()): env = wrapper_class(env, pufferlib.namespace(**[x for x in cfg.values()][0])) return pufferlib.emulation.GymnasiumPufferEnv( env=env, postprocessor_cls=pufferlib.emulation.BasicPostprocessor