undo

Author: TingquanGao

paddlex/inference/components/base.py

@@ -182,7 +182,7 @@ def _check_output(self, output, ori_data):
                     output_list.append(data)
                 return output_list
         else:
-            # assert isinstance(ori_data, dict) and isinstance(output, dict)
+            assert isinstance(ori_data, dict) and isinstance(output, dict)
             data = ori_data.copy() if self.keep_input else {}
             if isinstance(self.outputs, type(None)):
                 logging.debug(
@@ -193,7 +193,7 @@ def _check_output(self, output, ori_data):
                 for k, v in self.outputs.items():
                     if k not in output:
                         raise Exception(
-                            f"The value of key ({k}) is needed add to Data. But not found in output of {self.__class__.__name__}: ({output})!"
+                            f"The value of key ({k}) is needed add to Data. But not found in output of {self.__class__.__name__}: ({output.keys()})!"
                         )
                     data.update({v: output[k]})
         return [data]