re-structure

This commit is contained in:
Yaojia Wang
2026-02-01 22:55:31 +01:00
parent 400b12a967
commit b602d0a340
176 changed files with 856 additions and 853 deletions

View File

@@ -8,7 +8,7 @@ from uuid import uuid4
import pytest
from inference.data.repositories.annotation_repository import AnnotationRepository
from backend.data.repositories.annotation_repository import AnnotationRepository
class TestAnnotationRepositoryCreate:

View File

@@ -9,7 +9,7 @@ from uuid import uuid4
import pytest
from inference.data.repositories.batch_upload_repository import BatchUploadRepository
from backend.data.repositories.batch_upload_repository import BatchUploadRepository
class TestBatchUploadCreate:

View File

@@ -8,7 +8,7 @@ from uuid import uuid4
import pytest
from inference.data.repositories.dataset_repository import DatasetRepository
from backend.data.repositories.dataset_repository import DatasetRepository
class TestDatasetRepositoryCreate:
@@ -300,7 +300,7 @@ class TestActiveTrainingTasks:
repo = DatasetRepository()
# Update task to running
from inference.data.repositories.training_task_repository import TrainingTaskRepository
from backend.data.repositories.training_task_repository import TrainingTaskRepository
task_repo = TrainingTaskRepository()
task_repo.update_status(str(sample_training_task.task_id), "running")

View File

@@ -10,8 +10,8 @@ from uuid import uuid4
import pytest
from sqlmodel import select
from inference.data.admin_models import AdminAnnotation, AdminDocument
from inference.data.repositories.document_repository import DocumentRepository
from backend.data.admin_models import AdminAnnotation, AdminDocument
from backend.data.repositories.document_repository import DocumentRepository
def ensure_utc(dt: datetime | None) -> datetime | None:
@@ -243,7 +243,7 @@ class TestDocumentRepositoryDelete:
assert result is True
# Verify annotation is also deleted
from inference.data.repositories.annotation_repository import AnnotationRepository
from backend.data.repositories.annotation_repository import AnnotationRepository
ann_repo = AnnotationRepository()
annotations = ann_repo.get_for_document(str(sample_document.document_id))

View File

@@ -9,7 +9,7 @@ from uuid import uuid4
import pytest
from inference.data.repositories.model_version_repository import ModelVersionRepository
from backend.data.repositories.model_version_repository import ModelVersionRepository
class TestModelVersionCreate:

View File

@@ -8,7 +8,7 @@ from datetime import datetime, timezone, timedelta
import pytest
from inference.data.repositories.token_repository import TokenRepository
from backend.data.repositories.token_repository import TokenRepository
class TestTokenCreate:

View File

@@ -9,7 +9,7 @@ from uuid import uuid4
import pytest
from inference.data.repositories.training_task_repository import TrainingTaskRepository
from backend.data.repositories.training_task_repository import TrainingTaskRepository
class TestTrainingTaskCreate: