o $i2 @sldZddlmZddlmZmZddlmZeZ ede defddZ edde d ed ed efd dZ d S)z.Registry of connector names for global access.)Any) ConnectorConnectorContext) OldAPIStacknameclscCs,|tvrdSt|tstd||t|<dS)zRegister 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)rrr Z/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/rllib/connectors/registry.pyregister_connector s    r NctxparamsreturncCs"|tvr td|t|||S)zGet 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__typingrray.rllib.connectors.connectorrrray.rllib.utils.annotationsrdictrstrr rr r r r s  "