Merge pull request #277 from huangshiyu13/main

- fix bugs: AttributeError: module 'openrl.envs' has no attribute 'Pe…
OpenRL-Lab · Dec 5, 2023 · 5d21f65 · 5d21f65
2 parents e8c6ee9 + 9a05e6f
commit 5d21f65
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/openrl/envs/common/registration.py b/openrl/envs/common/registration.py
@@ -20,6 +20,7 @@
 import gymnasium as gym
 
 import openrl
+from openrl.envs.PettingZoo.registration import pettingzoo_env_dict
 from openrl.envs.vec_env import (
     AsyncVectorEnv,
     BaseVecEnv,
@@ -148,10 +149,7 @@ def make(
                 render_mode=convert_render_mode,
                 **kwargs,
             )
-        elif (
-            id in openrl.envs.pettingzoo_all_envs
-            or id in openrl.envs.PettingZoo.registration.pettingzoo_env_dict.keys()
-        ):
+        elif id in openrl.envs.pettingzoo_all_envs or id in pettingzoo_env_dict.keys():
             from openrl.envs.PettingZoo import make_PettingZoo_envs
 
             env_fns = make_PettingZoo_envs(