Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
3635acb
1
Parent(s):
48b9b47
Clean up imports and remove unused code across API modules
Browse files- api/dependencies.py +1 -1
- api/management.py +0 -2
- api/router.py +1 -7
api/dependencies.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
import os
|
2 |
import logging
|
3 |
from typing import Optional
|
4 |
from fastapi import Header, HTTPException
|
|
|
1 |
+
# import os # Supprimé car inutilisé
|
2 |
import logging
|
3 |
from typing import Optional
|
4 |
from fastapi import Header, HTTPException
|
api/management.py
CHANGED
@@ -5,7 +5,6 @@ from fastapi import APIRouter, HTTPException, Depends
|
|
5 |
from api.dependencies import verify_management_api_key
|
6 |
from db.models import fetch_model_by_id
|
7 |
from models.loader import model_pipelines, _load_single_model_pipeline
|
8 |
-
from models.schemas import ModelInfo
|
9 |
|
10 |
logger = logging.getLogger(__name__)
|
11 |
|
@@ -77,4 +76,3 @@ async def update_single_model(model_db_id: Any):
|
|
77 |
if model_db_id in model_pipelines:
|
78 |
del model_pipelines[model_db_id] # Assurer le déchargement en cas d'erreur imprévue
|
79 |
raise HTTPException(status_code=500, detail=f"Internal server error while updating model {model_db_id}. Model has been unloaded.")
|
80 |
-
|
|
|
5 |
from api.dependencies import verify_management_api_key
|
6 |
from db.models import fetch_model_by_id
|
7 |
from models.loader import model_pipelines, _load_single_model_pipeline
|
|
|
8 |
|
9 |
logger = logging.getLogger(__name__)
|
10 |
|
|
|
76 |
if model_db_id in model_pipelines:
|
77 |
del model_pipelines[model_db_id] # Assurer le déchargement en cas d'erreur imprévue
|
78 |
raise HTTPException(status_code=500, detail=f"Internal server error while updating model {model_db_id}. Model has been unloaded.")
|
|
api/router.py
CHANGED
@@ -1,15 +1,9 @@
|
|
1 |
import logging
|
2 |
import os
|
3 |
-
from
|
4 |
-
from fastapi import APIRouter, HTTPException, Depends, Header, Request, Response
|
5 |
|
6 |
-
# Import des dépendances
|
7 |
-
from api.dependencies import verify_api_key, verify_management_api_key
|
8 |
# Import des sous-routeurs
|
9 |
from api import prediction, management
|
10 |
-
from config.settings import MANAGEMENT_API_KEY
|
11 |
-
from db.models import fetch_model_by_id
|
12 |
-
from models.loader import model_pipelines, _load_single_model_pipeline, get_model
|
13 |
|
14 |
logger = logging.getLogger(__name__)
|
15 |
|
|
|
1 |
import logging
|
2 |
import os
|
3 |
+
from fastapi import APIRouter, Request, Response
|
|
|
4 |
|
|
|
|
|
5 |
# Import des sous-routeurs
|
6 |
from api import prediction, management
|
|
|
|
|
|
|
7 |
|
8 |
logger = logging.getLogger(__name__)
|
9 |
|