o `Û·i2ã @sldZddlmZddlmZmZddlmZeƒZ ede defdd„ƒZ edde d ed ed efd d„ƒZ d S)z.Registry of connector names for global access.é)ÚAny)Ú ConnectorÚConnectorContext)Ú OldAPIStackÚnameÚclscCs,|tvrdSt|tƒstd|ƒ‚|t|<dS)zƒRegister a connector for use with RLlib. Args: name: Name to register. cls: Callable that creates an env. Nz!Can only register Connector type.)ÚALL_CONNECTORSÚ issubclassrÚ TypeError)rr©r úS/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/rllib/connectors/registry.pyÚregister_connector s    r NÚctxÚparamsÚreturncCs"|tvr td|ƒ‚t| ||¡S)zìGet a connector by its name and serialized config. Args: name: name of the connector. ctx: Connector context. params: serialized parameters of the connector. Returns: Constructed connector. zconnector not found.)rÚ NameErrorÚ from_state)rrrr r r Ú get_connectors r)N) Ú__doc__ÚtypingrÚray.rllib.connectors.connectorrrÚray.rllib.utils.annotationsrÚdictrÚstrr rr r r r Ús  "