Fix bug

paddlehub/commands/run.py

@@ -74,7 +74,7 @@ class RunCommand(BaseCommand):
         return hub.Module(directory=module_dir[0])
 
     def add_module_config_arg(self):
-        configs = self.module.configs()
+        configs = self.module.processor.configs()
         for config in configs:
             if not config["dest"].startswith("--"):
                 config["dest"] = "--%s" % config["dest"]
@@ -104,7 +104,7 @@ class RunCommand(BaseCommand):
 
     def add_module_input_arg(self):
         module_type = self.module.type.lower()
-        expect_data_format = self.module.data_format(
+        expect_data_format = self.module.processor.data_format(
             self.module.default_signature)
         self.arg_input_group.add_argument(
             '--input_file',
@@ -144,14 +144,14 @@ class RunCommand(BaseCommand):
         if self.args.config:
             yaml_config = yaml_parser.parse(self.args.config)
         module_config = yaml_config.get("config", {})
-        for _config in self.module.configs():
+        for _config in self.module.processor.configs():
             key = _config['dest']
             module_config[key] = self.args.__dict__[key]
         return module_config
 
     def get_data(self):
         module_type = self.module.type.lower()
-        expect_data_format = self.module.data_format(
+        expect_data_format = self.module.processor.data_format(
             self.module.default_signature)
         input_data = {}
         if len(expect_data_format) == 1:
@@ -176,7 +176,7 @@ class RunCommand(BaseCommand):
         return input_data
 
     def check_data(self, data):
-        expect_data_format = self.module.data_format(
+        expect_data_format = self.module.processor.data_format(
             self.module.default_signature)
 
         if len(data.keys()) != len(expect_data_format.keys()):
@@ -236,10 +236,13 @@ class RunCommand(BaseCommand):
             return False
 
         # If the module is not executable, give an alarm and exit
-        if not self.module.default_signature:
+        if not self.module.is_runable:
             print("ERROR! Module %s is not executable." % module_name)
             return False
 
+        if self.module.code_version == "v2":
+            result = self.module(argv[1:])
+        else:
             self.module.check_processor()
             self.add_module_config_arg()
             self.add_module_input_arg()

paddlehub/module/module.py

@@ -137,6 +137,7 @@ class Module(object):
                  version=None):
         if not directory:
             return
+        self._code_version = "v2"
         self._directory = directory
         self.module_desc_path = os.path.join(self.directory, MODULE_DESC_PBNAME)
         self._desc = module_desc_pb2.ModuleDesc()
@@ -225,6 +226,14 @@ class Module(object):
     def name_prefix(self):
         return self._name_prefix
 
+    @property
+    def code_version(self):
+        return self._code_version
+
+    @property
+    def is_runable(self):
+        return False
+
 
 class ModuleHelper(object):
     def __init__(self, directory):
@@ -252,6 +261,7 @@ class ModuleV1(Module):
         if not directory:
             return
         super(ModuleV1, self).__init__(name, directory, module_dir, version)
+        self._code_version = "v1"
         self.program = None
         self.assets = []
         self.helper = None
@@ -501,11 +511,9 @@ class ModuleV1(Module):
         if not self.processor:
             raise ValueError("This Module is not callable!")
 
-    def configs(self):
-        return self.processor.configs()
-
-    def data_format(self, signature):
-        return self.processor.data_format(signature)
+    @property
+    def is_runable(self):
+        return self.default_signature != None
 
     def context(self,
                 sign_name=None,