o .i@s.ddlmZmZededZGdddZdS))AnyTypeVar_T)boundc@s:eZdZdZd ddZdefddZd edefd d ZdS) ExtensionManageraj A registry for managing pluggable extension classes. This class provides a simple mechanism to register and instantiate extension classes by name. It is commonly used to implement plugin systems where different implementations can be swapped at runtime. Examples: Basic usage with a registry instance: >>> FOO_REGISTRY = ExtensionManager() >>> @FOO_REGISTRY.register("my_foo_impl") ... class MyFooImpl(Foo): ... def __init__(self, value): ... self.value = value >>> foo_impl = FOO_REGISTRY.load("my_foo_impl", value=123) returnNcCs i|_dS)z9 Initialize an empty extension registry. N name2class)selfr P/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/utils/registry.py__init__s zExtensionManager.__init__namecsdtdtffdd }|S)zD Decorator to register a class with the given name. cls_to_registerrcs|j<|S)Nr)rrr r r wrap's z'ExtensionManager.register..wrap)r)r rrr rr register"szExtensionManager.registercls_namecOs2|j|}|dusJd|d||i|S)zN Instantiate and return a registered extension class by name. NzExtension class z not found)r get)r rargskwargsclsr r r load-s zExtensionManager.load)rN) __name__ __module__ __qualname____doc__r strrrrr r r r rs   rN)typingrrtyperrr r r r s