From b0b253ecdf41a0ed15535736ac71db849a3a700a Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Wed, 7 Apr 2021 12:00:02 +0900
Subject: [PATCH] Add data factory

---
 app/api/views/upload/factory.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/app/api/views/upload/factory.py b/app/api/views/upload/factory.py
index 14c3132c..2d1a8652 100644
--- a/app/api/views/upload/factory.py
+++ b/app/api/views/upload/factory.py
@@ -1,7 +1,15 @@
+from . import data
 from . import dataset
 from . import label
 
 
+def get_data_class(project_type: str):
+    if project_type in ['DocumentClassification', 'SequenceLabeling', 'Seq2seq']:
+        return data.TextData
+    else:
+        return data.FileData
+
+
 def get_dataset_class(format: str):
     if format == 'csv':
         return dataset.CsvDataset