Remove empty label name

app/api/views/upload/dataset.py

@@ -33,7 +33,7 @@ class Record:
 
     @property
     def label(self):
-        return [label.name for label in self._label if label.has_name()]
+        return [label.name for label in self._label if label.has_name() and label.name]
 
 
 class Records:

app/api/views/upload/label.py

@@ -1,5 +1,5 @@
 import abc
-from typing import Any, Dict
+from typing import Any, Dict, Union
 
 from pydantic import BaseModel
 
@@ -28,7 +28,7 @@ class Label(BaseModel, abc.ABC):
 
 
 class CategoryLabel(Label):
-    label: str
+    label: Union[str, int]
 
     def has_name(self) -> bool:
         return True
@@ -44,12 +44,12 @@ class CategoryLabel(Label):
         raise TypeError(f'{obj} is not str.')
 
     def replace(self, mapping: Dict[str, int]) -> 'Label':
-        label = mapping.get(self.label, self.label)
+        label = mapping[self.label]
         return CategoryLabel(label=label)
 
 
 class OffsetLabel(Label):
-    label: str
+    label: Union[str, int]
     start_offset: int
     end_offset: int
 
@@ -72,7 +72,7 @@ class OffsetLabel(Label):
             raise TypeError(f'{obj} is invalid type.')
 
     def replace(self, mapping: Dict[str, int]) -> 'Label':
-        label = mapping.get(self.label, self.label)
+        label = mapping[self.label]
         return OffsetLabel(
             label=label,
             start_offset=self.start_offset,

app/api/views/upload/labels.py

@@ -11,7 +11,13 @@ class Labels:
     def replace_label(self, mapping: Optional[Dict[str, int]] = None):
         if not mapping:
             return self
-        labels = [label.replace(mapping) for label in self.labels]
+        labels = []
+        for label in self.labels:
+            try:
+                label = label.replace(mapping)
+                labels.append(label)
+            except KeyError:
+                pass
         return Labels(labels)
 
     def dict(self) -> List[dict]: