diff --git a/backend/data_export/celery_tasks.py b/backend/data_export/celery_tasks.py index 167bf93b..63a40283 100644 --- a/backend/data_export/celery_tasks.py +++ b/backend/data_export/celery_tasks.py @@ -13,7 +13,7 @@ logger = get_task_logger(__name__) @shared_task def export_dataset(project_id, file_format: str, export_approved=False): project = get_object_or_404(Project, pk=project_id) - repository = create_repository(project) + repository = create_repository(project, file_format) writer = create_writer(file_format)(settings.MEDIA_ROOT) service = ExportApplicationService(repository, writer) filepath = service.export(export_approved) diff --git a/backend/data_export/pipeline/factories.py b/backend/data_export/pipeline/factories.py index 44a38235..9096a442 100644 --- a/backend/data_export/pipeline/factories.py +++ b/backend/data_export/pipeline/factories.py @@ -11,8 +11,8 @@ from projects.models import ( ) -def create_repository(project): - if getattr(project, "use_relation", False): +def create_repository(project, file_format: str): + if getattr(project, "use_relation", False) and file_format == catalog.JSONLRelation.name: return repositories.RelationExtractionRepository(project) mapping = { DOCUMENT_CLASSIFICATION: repositories.TextClassificationRepository,