update python-api.yml (#1)

* update python-api.yml * update requirements.txt * add model.pkl
AuHoh · Jul 4, 2023 · 155c3f9 · 155c3f9
1 parent 3226d4f
commit 155c3f9
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 3 deletions.
diff --git a/.github/workflows/python-api.yml b/.github/workflows/python-api.yml
@@ -29,4 +29,4 @@ jobs:
         if [ -f api/requirements.txt ]; then pip install -r api/requirements.txt; fi
     - name: Test API with unit test
       run: |
-        python -m unittest discover -s api -t api/tests
+        export PYTHONPATH=api MODEL_PATH=api/tests/model.pkl && python -m unittest discover -s api/tests -t api
diff --git a/api/main.py b/api/main.py
@@ -1,5 +1,6 @@
 import json
 import pickle
+import os
 from smart_open import open
 
 import pandas as pd
@@ -8,7 +9,7 @@
 from sample_call import test_dict
 
 th_proba = 0.54
-model_path = '/Users/audreyhohmann/Documents/Formation/OCR/P7/mlruns/274439274391636157/99fce0a77f4045eda1e8cb7e14c57ab7/artifacts/model/model.pkl'
+model_path = os.getenv('MODEL_PATH', 'tests/model.pkl')
 with open(model_path, 'rb') as file:
     model = pickle.load(file)
 

diff --git a/api/requirements.txt b/api/requirements.txt
@@ -1,2 +1,5 @@
 fastapi==0.99.1
-pandas==1.5.3
+pandas==1.5.3
+httpx==0.24.1
+smart-open==6.3.0
+lightgbm==3.3.5
diff --git a/api/tests/model.pkl b/api/tests/model.pkl