def load_module_info(cls, directory: str) -> EasyDict:
'''Load the infomation of Module object defined in the specified directory.'''
desc_file = os.path.join(directory, 'module_desc.pb')
desc = module_v1_utils.convert_module_desc(desc_file)
# The naming of some old versions of Module is not standardized, which format of uppercase
# letters. This will cause the path of these modules to be incorrect after installation.
module_info = desc.module_info
module_info.name = module_info.name.lower()
return module_info
def __init__(self, name: str = None, directory: str = None, version: str = None):
if not directory:
return
desc_file = os.path.join(directory, 'module_desc.pb')
self.desc = module_v1_utils.convert_module_desc(desc_file)
self.helper = self
self.signatures = self.desc.signatures
self.default_signature = self.desc.default_signature
self.directory = directory
self._load_model()
self._load_parameters()
self._load_processor()
self._load_assets()
self._load_extra_info()
self._generate_func()