From ce342854a79eba4603771c0dc555e7a75aab4f2d Mon Sep 17 00:00:00 2001 From: wuzewu Date: Thu, 10 Sep 2020 16:29:59 +0800 Subject: [PATCH] Add get_py_requirments in module --- paddlehub/module/module.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/paddlehub/module/module.py b/paddlehub/module/module.py index cf620a92..22e99611 100644 --- a/paddlehub/module/module.py +++ b/paddlehub/module/module.py @@ -57,12 +57,13 @@ def serving(func: Callable) -> Callable: class Module(object): def __new__(cls, name: str = None, directory: str = None, version: str = None, **kwargs): if cls.__name__ == 'Module': + # This branch come from hub.Module(name='xxx') if name: module = cls.init_with_name(name=name, version=version, **kwargs) elif directory: module = cls.init_with_directory(directory=directory, **kwargs) else: - raise RuntimeError() + module = object.__new__(cls) module.directory = directory return module @@ -105,6 +106,15 @@ class Module(object): user_module_cls = cls.load(directory) return user_module_cls(**kwargs) + @classmethod + def get_py_requirments(cls): + req_file = os.path.join(cls.directory, 'requirements.txt') + if not os.path.exists(req_file): + return [] + + with open(req_file, 'r') as file: + return file.read().split('\n') + class RunModule(object): def __init__(self, *args, **kwargs): -- GitLab