kai/invoice-master-poc-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

re-structure

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/web/test_dataset_routes.py
View File

@@ -9,9 +9,9 @@ from datetime import datetime, timezone
from unittest.mock import MagicMock, patch
from uuid import UUID
from inference.data.admin_models import TrainingDataset, DatasetDocument
from inference.web.api.v1.admin.training import create_training_router
from inference.web.schemas.admin import (
from backend.data.admin_models import TrainingDataset, DatasetDocument
from backend.web.api.v1.admin.training import create_training_router
from backend.web.schemas.admin import (
DatasetCreateRequest,
DatasetTrainRequest,
TrainingConfig,
@@ -130,10 +130,10 @@ class TestCreateDatasetRoute:
)
with patch(
"inference.web.services.dataset_builder.DatasetBuilder",
"backend.web.services.dataset_builder.DatasetBuilder",
return_value=mock_builder,
), patch(
"inference.web.api.v1.admin.training.datasets.get_storage_helper"
"backend.web.api.v1.admin.training.datasets.get_storage_helper"
) as mock_storage:
mock_storage.return_value.get_datasets_base_path.return_value = "/data/datasets"
mock_storage.return_value.get_admin_images_base_path.return_value = "/data/admin_images"
@@ -222,10 +222,10 @@ class TestCreateDatasetRoute:
)
with patch(
"inference.web.services.dataset_builder.DatasetBuilder",
"backend.web.services.dataset_builder.DatasetBuilder",
return_value=mock_builder,
), patch(
"inference.web.api.v1.admin.training.datasets.get_storage_helper"
"backend.web.api.v1.admin.training.datasets.get_storage_helper"
) as mock_storage:
mock_storage.return_value.get_datasets_base_path.return_value = "/data/datasets"
mock_storage.return_value.get_admin_images_base_path.return_value = "/data/admin_images"