re-structure
This commit is contained in:
@@ -11,7 +11,7 @@ import pytest
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from inference.data.admin_models import (
|
||||
from backend.data.admin_models import (
|
||||
AdminAnnotation,
|
||||
AdminDocument,
|
||||
AdminToken,
|
||||
@@ -20,8 +20,8 @@ from inference.data.admin_models import (
|
||||
TrainingDataset,
|
||||
TrainingTask,
|
||||
)
|
||||
from inference.web.api.v1.admin.dashboard import create_dashboard_router
|
||||
from inference.web.core.auth import get_admin_token_dep
|
||||
from backend.web.api.v1.admin.dashboard import create_dashboard_router
|
||||
from backend.web.core.auth import validate_admin_token
|
||||
|
||||
|
||||
def create_test_app(override_token_dep):
|
||||
@@ -31,7 +31,7 @@ def create_test_app(override_token_dep):
|
||||
app.include_router(router)
|
||||
|
||||
# Override auth dependency
|
||||
app.dependency_overrides[get_admin_token_dep] = lambda: override_token_dep
|
||||
app.dependency_overrides[validate_admin_token] = lambda: override_token_dep
|
||||
|
||||
return app
|
||||
|
||||
|
||||
Reference in New Issue
Block a user