o c²“iÁã@s¤ddlZddlmZmZddlZddlmZddlm Z m Z ddl m Z e  e¡Ze ddefdd „ƒZe ddefd d „ƒZd ed eeeejfdejfdd„ZdS)éN)ÚTypeÚUnion)Ú EnvContext)ÚERR_MSG_INVALID_ENV_DESCRIPTORÚEnvError)Ú PublicAPIFÚerrorcCó0zddl}|WSty|rtdƒ‚YdSw)a Tries importing pyspiel and returns the module (or None). Args: error: Whether to raise an error if pyspiel cannot be imported. Returns: The pyspiel module. Raises: ImportError: If error=True and pyspiel is not installed. rNzˆCould not import pyspiel! Pygame is not a dependency of RLlib and RLlib requires you to install pygame separately: `pip install pygame`.)ÚpyspielÚ ImportError)rr ©r úP/home/ubuntu/.local/lib/python3.10/site-packages/ray/rllib/env/utils/__init__.pyÚtry_import_pyspieló  ÿùrcCr )aTries importing open_spiel and returns the module (or None). Args: error: Whether to raise an error if open_spiel cannot be imported. Returns: The open_spiel module. Raises: ImportError: If error=True and open_spiel is not installed. rNz—Could not import open_spiel! open_spiel is not a dependency of RLlib and RLlib requires you to install open_spiel separately: `pip install open_spiel`.)Ú open_spielr )rrr r r Útry_import_open_spiel,rrÚ env_contextÚenv_descriptorÚreturnc Cs¦z ddl}| ¡Wn tttfyYnwt|tƒr*| d¡r*ddl}t   |¡zt|t ƒr7||ƒ}W|St j |fi|¤Ž}W|St j jyRtt |¡ƒ‚w)a¿Tries to create a gym env given an EnvContext object and descriptor. Note: This function tries to construct the env from a string descriptor only using possibly installed RL env packages (such as gym, pybullet_envs, etc). These packages are no installation requirements for RLlib. In case you would like to support more such env packages, add the necessary imports and construction logic below. Args: env_context: The env context object to configure the env. Note that this is a config dict, plus the properties: `worker_index`, `vector_index`, and `remote`. env_descriptor: The env descriptor as a gym-registered string, e.g. CartPole-v1, ALE/MsPacman-v5, or CartPoleContinuousBulletEnv-v0. Alternatively, the gym.Env subclass to use. Returns: The actual gym environment object. Raises: gym.error.Error: If the env cannot be constructed. rNz ale_py:ALE/)Ú pybullet_envsÚgetListÚAttributeErrorÚModuleNotFoundErrorr Ú isinstanceÚstrÚ startswithÚale_pyÚgymÚ register_envsÚtypeÚmakerÚErrorrrÚformat)rrrrÚenvr r r Ú_gym_env_creatorGs$ ÿ   üýÿr$)F)ÚloggingÚtypingrrÚ gymnasiumrÚray.rllib.env.env_contextrÚray.rllib.utils.errorrrÚray.util.annotationsrÚ getLoggerÚ__name__ÚloggerÚboolrrrÚEnvr$r r r r Ús$   ÿþý