re-structure

tests/integration/services/test_dashboard_service_integration.py

@@ -9,7 +9,7 @@ from uuid import uuid4
 
 import pytest
 
-from inference.data.admin_models import (
+from backend.data.admin_models import (
     AdminAnnotation,
     AdminDocument,
     AnnotationHistory,
@@ -17,7 +17,7 @@ from inference.data.admin_models import (
     TrainingDataset,
     TrainingTask,
 )
-from inference.web.services.dashboard_service import (
+from backend.web.services.dashboard_service import (
     DashboardStatsService,
     DashboardActivityService,
     is_annotation_complete,

tests/integration/services/test_dataset_builder_integration.py

@@ -12,11 +12,11 @@ from uuid import uuid4
 import pytest
 import yaml
 
-from inference.data.admin_models import AdminAnnotation, AdminDocument
-from inference.data.repositories.annotation_repository import AnnotationRepository
-from inference.data.repositories.dataset_repository import DatasetRepository
-from inference.data.repositories.document_repository import DocumentRepository
-from inference.web.services.dataset_builder import DatasetBuilder
+from backend.data.admin_models import AdminAnnotation, AdminDocument
+from backend.data.repositories.annotation_repository import AnnotationRepository
+from backend.data.repositories.dataset_repository import DatasetRepository
+from backend.data.repositories.document_repository import DocumentRepository
+from backend.web.services.dataset_builder import DatasetBuilder
 
 
 @pytest.fixture

tests/integration/services/test_document_service_integration.py

@@ -9,7 +9,7 @@ from unittest.mock import MagicMock
 
 import pytest
 
-from inference.web.services.document_service import DocumentService, DocumentResult
+from backend.web.services.document_service import DocumentService, DocumentResult
 
 
 class MockStorageBackend: