remove type reader

paddle_hub/commands/run.py

@@ -108,9 +108,7 @@ class RunCommand(BaseCommand):
         origin_data = csv_reader.read(self.args.dataset)
         input_data = {}
         for key, value in yaml_config['input_data'].items():
-            type_reader = hub.DataType.type_reader(value['type'])
-            input_data[key] = list(
-                map(type_reader.read, origin_data[value['key']]))
+            input_data[key] = origin_data[value['key']]
 
         # run module with data
         config = yaml_config.get("config", {})

paddle_hub/io/type.py

@@ -55,34 +55,3 @@ class DataType(Enum):
     def is_valid_type(cls, data_type):
         data_type = DataType.type(data_type)
         return data_type in DataType
-
-    @classmethod
-    def type_reader(cls, data_type):
-        data_type = DataType.type(data_type)
-        if not DataType.is_valid_type(data_type):
-            logger.critical("invalid data type %s" % data_type)
-            exit(1)
-
-        if data_type == DataType.IMAGE:
-            return ImageReader
-        elif data_type == DataType.TEXT:
-            return TextReader
-        else:
-            type_str = DataType.str(data_type)
-            logger.critical(
-                "data type %s not supported for the time being" % type_str)
-            exit(1)
-
-
-class ImageReader:
-    @classmethod
-    def read(cls, path):
-        utils.check_path(path)
-        image = Image.open(path)
-        return image
-
-
-class TextReader:
-    @classmethod
-    def read(cls, text):
-        return text