From e869e7eaa14ce53034f07234a0cc15210f7625d4 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Mon, 5 Apr 2021 11:46:56 +0900
Subject: [PATCH] Update dataset

---
 app/api/views/upload/dataset.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/api/views/upload/dataset.py b/app/api/views/upload/dataset.py
index fa0f7789..ead819b9 100644
--- a/app/api/views/upload/dataset.py
+++ b/app/api/views/upload/dataset.py
@@ -19,6 +19,9 @@ class Record:
         self.label = label
         self.metadata = metadata
 
+    def __str__(self):
+        return f'{self.data}\t{self.label}'
+
 
 class Dataset:
 
@@ -46,7 +49,8 @@ class Dataset:
 
     def from_row(self, filename: str, row: Dict) -> Record:
         data = row.pop(self.column_data)
-        label = row.pop(self.column_label)
+        label = row.pop(self.column_label, [])
+        label = [label] if isinstance(label, str) else label
         record = Record(filename=filename, data=data, label=label, metadata=row)
         return record