From c81930624b4d2056ea6319715b08b9d2aac8eca1 Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Tue, 12 Mar 2019 19:27:28 +0800
Subject: [PATCH] add TEXT type support

---
 paddle_hub/data/type.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/paddle_hub/data/type.py b/paddle_hub/data/type.py
index c4d99253..02039981 100644
--- a/paddle_hub/data/type.py
+++ b/paddle_hub/data/type.py
@@ -65,6 +65,8 @@ class DataType(Enum):
 
         if data_type == DataType.IMAGE:
             return ImageReader
+        elif data_type == DataType.TEXT:
+            return TextReader
         else:
             type_str = DataType.str(data_type)
             logger.critical(
@@ -78,3 +80,9 @@ class ImageReader:
         utils.check_path(path)
         image = Image.open(path)
         return image
+
+
+class TextReader:
+    @classmethod
+    def read(cls, text):
+        return text
-- 
GitLab