- fix bugs: AttributeError: module 'openrl.envs' has no attribute 'PettingZoo'

Author: huangshiyu13

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,
@@ -29,7 +30,7 @@
 )
 from openrl.envs.vec_env.vec_info import VecInfoFactory
 from openrl.rewards import RewardFactory
-from openrl.envs.PettingZoo.registration import pettingzoo_env_dict
+
 
 def make(
     id: str,
@@ -148,10 +149,7 @@ def make(
                 render_mode=convert_render_mode,
                 **kwargs,
             )
-        elif (
-            id in openrl.envs.pettingzoo_all_envs
-            or id in 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(