Merge pull request #249 from ShenYuhan/add_module_dir_version

add module_dir version

Merge pull request #249 from ShenYuhan/add_module_dir_version
add module_dir version
e5e8afc3 · Bin Long · GitHub · f68214c9 · b5ccd7fe · e5e8afc3
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

paddlehub/module/module.py paddlehub/module/module.py +4 -1

未找到文件。
--- a/paddlehub/module/module.py
+++ b/paddlehub/module/module.py
@@ -129,7 +129,10 @@ class Module(object):
            self._init_with_module_file(module_dir=module_dir[0])
            lock.flock(fp_lock, lock.LOCK_UN)
            name = module_dir[0].split("/")[-1]
-            version = module_dir[1]
+            if len(module_dir) > 1:
+                version = module_dir[1]
+            else:
+                version = default_module_manager.search_module(name)[1]
        elif signatures:
            if processor:
                if not issubclass(processor, BaseProcessor):