diff --git a/requirements.txt b/requirements.txt index f689ba5..579fe9b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,21 +1,3 @@ -fastapi==0.88.0 -pydantic==1.10.2 -uvicorn==0.20.0 - -# local requirements -# uncomment to run the system locally -# torch==1.13.0 -# jaqpotpy==1.0.82 -# tqdm==4.64.1 -# skl2onnx==1.13.0 -# onnxruntime==1.13.1 -# matplotlib==3.6.2 -# numpy~=1.26.4 -# networkx~=2.8.8 -# pillow~=10.3.0 -# scipy~=1.13.0 -# torch-scatter -# torch-sparse -# torch-cluster -# torch-spline-conv -# torch-geometric +fastapi==0.111.0 +pydantic==2.7.1 +uvicorn==0.29.0 \ No newline at end of file diff --git a/src/entities/prediction_request.py b/src/entities/prediction_request.py index 0846a9f..69767a7 100644 --- a/src/entities/prediction_request.py +++ b/src/entities/prediction_request.py @@ -15,4 +15,4 @@ class PredictionRequestPydantic(BaseModel): dataset: Any rawModel: Any additionalInfo: Any - doaMatrix: Any + doaMatrix: Any = None \ No newline at end of file