One more configuration try

MLOps-essi-upc · Dec 12, 2023 · aee3b16 · aee3b16
1 parent a6f5b54
commit aee3b16
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/backend.yml b/.github/workflows/backend.yml
@@ -38,7 +38,7 @@
          echo "MLFLOW_TRACKING_PASSWORD=${{ secrets.MLFLOW_TRACKING_PASSWORD }}" >> .env
 
      - name: Install packages # Install dependencies needed for tests (not for whole project)
-       run: pip install pytest mlflow python-dotenv fastapi httpx uvicorn
+       run: pip install pytest mlflow python-dotenv fastapi httpx torch torchvision pydantic
 
      - name: Run tests
        run: |

diff --git a/src/app/api.py b/src/app/api.py
@@ -2,7 +2,6 @@
 from typing import List
 from PIL import Image
 from io import BytesIO
-import uvicorn
 from fastapi import FastAPI, Request, File, UploadFile, HTTPException
 from torchvision import transforms
 import torch