fixed error

thatguy11325 · Mar 12, 2024 · dfb5b44 · dfb5b44
1 parent 9f5c968
commit dfb5b44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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