Browse Source

Enable to pass mypy

pull/1764/head
Hironsan 2 years ago
parent
commit
5ad4bb777c
1 changed files with 4 additions and 5 deletions
  1. 9
      backend/data_export/pipeline/repositories.py

9
backend/data_export/pipeline/repositories.py

@ -1,7 +1,7 @@
import abc import abc
import itertools import itertools
from collections import defaultdict from collections import defaultdict
from typing import Any, Dict, Iterator, List, Tuple, Union
from typing import Any, Dict, Iterator, List, Tuple
from .data import Record from .data import Record
from examples.models import Example from examples.models import Example
@ -10,13 +10,12 @@ from projects.models import Project
SpanType = Tuple[int, int, str] SpanType = Tuple[int, int, str]
class BaseRepository(abc.ABC):
class BaseRepository:
def __init__(self, project: Project): def __init__(self, project: Project):
self.project = project self.project = project
@abc.abstractmethod
def list(self, export_approved=False) -> Iterator[Record]: def list(self, export_approved=False) -> Iterator[Record]:
pass
raisen> <span class="ne">NotImplementedError()
class FileRepository(BaseRepository): class FileRepository(BaseRepository):
@ -192,7 +191,7 @@ class IntentDetectionSlotFillingRepository(TextRepository):
def label_per_user(self, doc) -> Dict: def label_per_user(self, doc) -> Dict:
category_per_user: Dict[str, List[str]] = defaultdict(list) category_per_user: Dict[str, List[str]] = defaultdict(list)
span_per_user: Dict[str, List[SpanType]] = defaultdict(list) span_per_user: Dict[str, List[SpanType]] = defaultdict(list)
label_per_user: Dict[str, Dict[str, Union[List[str], List[SpanType]]]] = defaultdict(dict)
label_per_user: Dict[str, Dict[str, List]] = defaultdict(dict)
for a in doc.categories.all(): for a in doc.categories.all():
category_per_user[a.user.username].append(a.label.text) category_per_user[a.user.username].append(a.label.text)
for a in doc.spans.all(): for a in doc.spans.all():

Loading…
Cancel
Save