Browse Source

Update dataset

pull/1310/head
Hironsan 3 years ago
parent
commit
e869e7eaa1
1 changed files with 5 additions and 1 deletions
  1. 6
      app/api/views/upload/dataset.py

6
app/api/views/upload/dataset.py

@ -19,6 +19,9 @@ class Record:
self.label = label self.label = label
self.metadata = metadata self.metadata = metadata
def __str__(self):
return f'{self.data}\t{self.label}'
class Dataset: class Dataset:
@ -46,7 +49,8 @@ class Dataset:
def from_row(self, filename: str, row: Dict) -> Record: def from_row(self, filename: str, row: Dict) -> Record:
data = row.pop(self.column_data) 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) record = Record(filename=filename, data=data, label=label, metadata=row)
return record return record

Loading…
Cancel
Save