Browse Source
Merge pull request #1723 from youichiro/fix_import_dataset
Bug fix: Specify project when retrieving labels in celery task
pull/1727/head
Hiroki Nakayama
2 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
1 additions and
1 deletions
-
backend/data_import/pipeline/writers.py
|
|
@ -70,7 +70,7 @@ class Examples: |
|
|
mapping = {} |
|
|
mapping = {} |
|
|
label_types: List[Type[LabelType]] = [CategoryType, SpanType] |
|
|
label_types: List[Type[LabelType]] = [CategoryType, SpanType] |
|
|
for model in label_types: |
|
|
for model in label_types: |
|
|
for label in model.objects.all(): |
|
|
|
|
|
|
|
|
for label in model.objects.filter(project=project): |
|
|
mapping[label.text] = label |
|
|
mapping[label.text] = label |
|
|
annotations = list( |
|
|
annotations = list( |
|
|
itertools.chain.from_iterable( |
|
|
itertools.chain.from_iterable( |
|
|
|
