From 97c645281ec1aa0a04c4edd184e1e3ad2ce1aea0 Mon Sep 17 00:00:00 2001
From: AntoinePELAMOURGUES <antoinepela@gmail.com>
Date: Tue, 19 Nov 2024 14:55:15 +0400
Subject: [PATCH] :hammer: Update tests files

---
 api/main.py                 |  4 ++--
 tests/requirements-test.txt | 21 ++++++++++++---------
 2 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/api/main.py b/api/main.py
index 9405601..f5153a1 100644
--- a/api/main.py
+++ b/api/main.py
@@ -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
 
diff --git a/tests/requirements-test.txt b/tests/requirements-test.txt
index 6e11381..d9ab17b 100644
--- a/tests/requirements-test.txt
+++ b/tests/requirements-test.txt
@@ -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
@@ -24,3 +25,5 @@ python-dotenv
 surprise
 mlflow
 
+streamlit
+