alexfremont commited on
Commit
3635acb
·
1 Parent(s): 48b9b47

Clean up imports and remove unused code across API modules

Browse files
Files changed (3) hide show
  1. api/dependencies.py +1 -1
  2. api/management.py +0 -2
  3. 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 typing import Any, Optional, Dict, List
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