o W+ i @srddlZddlZddlmZmZmZddlmZdZdZ eZ Gddde Z e dfde d e d ed e fd d ZdS)N)ListTupleUnion) get_loggertypedefaultc@seZdZdZdefddZddZeddZed d Z d d Z e fd dZ e dddfddZ e dddfdededefddZdS)Registryz Registry which support registering modules and group them by a keyname If group name is not provided, modules will be registered to default group. namecCs||_tii|_dSN)_name default_group_modules)selfr re/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/modelscope/utils/registry.py__init__szRegistry.__init__cCsL|jjd|jd}|jD]\}}|d|dt|d7}q|S)Nz (z) group_name=z , modules= ) __class____name__r r itemslistkeys)r format_str group_namegrouprrr__repr__s    zRegistry.__repr__cC|jSr )r rrrrr z Registry.namecCrr )r rrrrmodules$rzRegistry.modulescCsP|jD] \}}td||D] }td|qtdqdS)z8 logging the list of module in current registry r N)r rloggerinfor)rrrmrrrr(s   z Registry.listcCs ||jvrdS|j||dSr )r get)rZ module_key group_keyrrrr&1s z Registry.getNFcCs|t|ts Jd||jvrt|j|<|dur|j}||j|vr2|s2t|d|jd|d||j||<||_dS)Nz%group_key is required and must be strz is already registered in []) isinstancestrr dictrKeyErrorr r')rr' module_name module_clsforcerrr_register_module7s"    zRegistry._register_moduler'r.r/csXdusttstdt|dur!j|d|Sfdd}|S)a Register module Example: >>> models = Registry('models') >>> @models.register_module('image-classification', 'SwinT') >>> class SwinTransformer: >>> pass >>> @models.register_module('SwinDefault') >>> class SwinTransformerDefaultGroup: >>> pass >>> class SwinTransformer2: >>> pass >>> MODELS.register_module('image-classification', module_name='SwinT2', module_cls=SwinTransformer2) Args: group_key: Group name of which module will be registered, default group name is 'default' module_name: Module name module_cls: Module class object force (bool, optional): Whether to override an existing class with the same name. Default: False. Nz,module_name must be either of None, str,got r'r.r/r0csj|d|S)Nr2)r1)r/r0r'r.rrr _register{sz+Registry.register_module.._register)r*r+ TypeErrorrr1)rr'r.r/r0r4rr3rregister_moduleOs zRegistry.register_module)r __module__ __qualname____doc__r+rrpropertyr r rr r&r1rr6rrrrrs4    rregistryr' default_argsreturnc Cst|tstdt|t|vr'|dust|vr'tdtd|d|t|ts5tdt|t|tsG|dusGtdt|dd lm}|j ||d f}| || }|durs| D] \}}|||qh|duryt}|t} t| tr|j| |d } | durt| d |j d |d|| _nt| st| r| } n tdt| zt| dr| jdi|WS| di|WSty} z t| | jd| | d} ~ ww)aBuild a module from config dict when it is a class configuration, or call a function from config dict when it is a function configuration. Example: >>> models = Registry('models') >>> @models.register_module('image-classification', 'SwinT') >>> class SwinTransformer: >>> pass >>> swint = build_from_cfg(dict(type='SwinT'), MODELS, >>> 'image-classification') >>> # Returns an instantiated object >>> >>> @MODELS.register_module() >>> def swin_transformer(): >>> pass >>> = build_from_cfg(dict(type='swin_transformer'), MODELS) >>> # Return a result of the calling function Args: cfg (dict): Config dict. It should at least contain the key "type". registry (:obj:`Registry`): The registry to search the type from. group_key (str, optional): The name of registry group from which module should be searched. default_args (dict, optional): Default initialization arguments. type_name (str, optional): The name of the type in the config. Returns: object: The constructed object. zcfg must be a dict, but got Nz.`cfg` or `default_args` must contain the key "z ", but got rz8registry must be an modelscope.Registry object, but got z-default_args must be a dict or None, but got r)LazyImportModuler)r'z is not in the z registry group zE. Please make sure the correct version of ModelScope library is used.z*type must be a str or valid type, but got _instantiatez: r)r*r,r5r TYPE_NAMEr-rZmodelscope.utils.import_utilsr>r upper import_modulecopyr setdefaultr popr+r&r'inspectisclass isfunctionhasattrr? Exceptionr) cfgr;r'r<r>sigargsr valueobj_typeZobj_clserrrbuild_from_cfgsf        rQ) importlibrFtypingrrrZmodelscope.utils.loggerrr@r r#objectrr+r,rQrrrrs& z