From 8507ec2d470b4fd34e39ee525de8b1d92268ccad Mon Sep 17 00:00:00 2001 From: wuzewu Date: Fri, 21 Feb 2020 20:16:31 +0800 Subject: [PATCH] Supports initialization of Module subclasses --- paddlehub/module/module.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/paddlehub/module/module.py b/paddlehub/module/module.py index 3ac4d0c9..59d52f94 100644 --- a/paddlehub/module/module.py +++ b/paddlehub/module/module.py @@ -163,7 +163,13 @@ class Module(object): module = cls.init_with_directory(directory=directory, **kwargs) CacheUpdater("update_cache", module.name, module.version).start() else: - module = object.__new__(cls) + if not name and not directory: + directory = os.path.dirname( + sys.modules[cls.__module__].__file__) + module = Module.init_with_directory( + directory=directory, **kwargs) + else: + module = object.__new__(cls) return module -- GitLab