Update factory for export repository

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)

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,