|
|
@ -0,0 +1,18 @@ |
|
|
|
from typing import Dict, List, Optional |
|
|
|
|
|
|
|
from .label import Label |
|
|
|
|
|
|
|
|
|
|
|
class Labels: |
|
|
|
|
|
|
|
def __init__(self, labels: List[Label]): |
|
|
|
self.labels = 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] |
|
|
|
return Labels(labels) |
|
|
|
|
|
|
|
def dict(self) -> List[dict]: |
|
|
|
return [label.dict() for label in self.labels] |