diff --git a/backend/data_export/pipeline/repositories.py b/backend/data_export/pipeline/repositories.py index e50eafe3..71420b7c 100644 --- a/backend/data_export/pipeline/repositories.py +++ b/backend/data_export/pipeline/repositories.py @@ -201,4 +201,7 @@ class IntentDetectionSlotFillingRepository(TextRepository): label_per_user[user]["cats"] = cats for user, span in span_per_user.items(): label_per_user[user]["entities"] = span + for label in label_per_user.values(): + label.setdefault("cats", []) + label.setdefault("entities", []) return label_per_user diff --git a/backend/data_export/tests/test_repositories.py b/backend/data_export/tests/test_repositories.py index 0ff25287..b0d29689 100644 --- a/backend/data_export/tests/test_repositories.py +++ b/backend/data_export/tests/test_repositories.py @@ -77,7 +77,7 @@ class TestSequenceLabelingRepository(TestRepository): def test_list(self): project = prepare_project(SEQUENCE_LABELING) - repository = SequenceLabelingRepository(project) + repository = SequenceLabelingRepository(project.item) self.prepare_data(project) expected = [ { @@ -95,7 +95,7 @@ class TestSequenceLabelingRepository(TestRepository): def test_list_on_collaborative_annotation(self): project = prepare_project(SEQUENCE_LABELING, collaborative_annotation=True) - repository = SequenceLabelingRepository(project) + repository = SequenceLabelingRepository(project.item) self.prepare_data(project) expected = [ {