Specify project when retrieving labels

3 years ago · 5de30a0774
1 changed files with 1 additions and 1 deletions
--- a/backend/data_import/pipeline/writers.py
+++ b/backend/data_import/pipeline/writers.py
@ -70,7 +70,7 @@ class Examples:
        mapping = {}
        label_types: List[Type[LabelType]] = [CategoryType, SpanType]
        for model in label_types:
-            for label in model.objects.all():
+            for label in model.objects.filter(project=project):
                mapping[label.text] = label
        annotations = list(
            itertools.chain.from_iterable(