Skip to content

Commit

Permalink
🔨 Update tests files
Browse files Browse the repository at this point in the history
  • Loading branch information
AntoinePELAMOURGUES committed Nov 19, 2024
1 parent 1cfe955 commit 97c6452
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 11 deletions.
4 changes: 2 additions & 2 deletions api/main.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from typing import Union
from fastapi import FastAPI, status, Depends, HTTPException
from typing import Annotated
from auth import get_current_user, router as auth_router
from predict import router as predict_router
from api.auth import get_current_user, router as auth_router
from api.predict import router as predict_router
from prometheus_fastapi_instrumentator import Instrumentator
import logging

Expand Down
21 changes: 12 additions & 9 deletions tests/requirements-test.txt
Original file line number Diff line number Diff line change
@@ -1,21 +1,22 @@
# Test framework
pytest==7.4.3
pytest

# Database testing
psycopg2-binary==2.9.9
psycopg2-binary

# API testing
fastapi==0.104.1
httpx==0.25.1
python-multipart==0.0.6
fastapi
httpx
python-multipart

# Data processing and ML dependencies
pandas==2.1.3
numpy==1.26.2
scikit-learn==1.3.2
pandas
numpy
scikit-learn

# Test client for FastAPI
requests==2.31.0
requests
passlib

# Test client for Airflow
apache-airflow
Expand All @@ -24,3 +25,5 @@ python-dotenv
surprise
mlflow

streamlit

0 comments on commit 97c6452

Please sign in to comment.