build: Upgrade stale packages (fastapi, starlette, tensorflow, torch) (…

…SeldonIO#1603) * use py >= 3.9 * update packages * fix tests due to fastapi `0.109.2` * upgrade mypy to fix issues after upgrading starlette * remove py3.8 from CI tests * add a min version for tritonclient * regenerate lock file
sakoush · Mar 1, 2024 · 05d6a0c · 05d6a0c
1 parent 7ac4351
commit 05d6a0c
Show file tree

Hide file tree

Showing 23 changed files with 11,992 additions and 18,733 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -35,7 +35,7 @@ jobs:
         os: [ubuntu-latest, macos-latest]
         # NOTE: There's no pre-built `grpcio` wheel for Python 3.11 yet
         # https://github.com/grpc/grpc/issues/32454
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.9", "3.10"]
         is-pr:
           - ${{ github.event_name == 'pull_request' }}
         exclude:
@@ -83,7 +83,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.9", "3.10"]
         tox-environment:
           - sklearn
           - xgboost
@@ -129,7 +129,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: [3.9", "3.10"]
     runs-on: ${{ matrix.os }}
     steps:
       - name: Maximize build space

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,7 +43,7 @@ asyncio_mode = "auto"
 addopts = "--import-mode=importlib"
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<3.12"
+python = ">=3.9,<3.12"
 click = "*"
 fastapi = ">=0.88.0,!=0.89.0,<=0.109.2"
 python-dotenv = "*"
@@ -55,14 +55,13 @@ uvicorn = "*"
 starlette-exporter = "*"
 py-grpc-prometheus = "*"
 aiokafka = "*"
-## Tritonclient has recently started requiring cuda-python, which doesn't work
-## in MacOS.
-## Adding a ceiling until this issue gets resolved:
-##    https://github.com/triton-inference-server/server/issues/6246
-tritonclient = {version = ">=2.24,<2.43", extras = ["http"]}
+# add a min version to tritonclient due to https://github.com/triton-inference-server/server/issues/6246
+tritonclient = ">=2.40"
 aiofiles = "*"
 orjson = "*"
 uvloop = {version = "*", markers = "sys_platform != 'win32' and (sys_platform != 'cygwin' and platform_python_implementation != 'PyPy')"}
+pydantic = "<2.0.0"
+python-multipart = "*"
 
 ## The importlib-resources backport is required to use some
 ## functionality added in Python 3.10
@@ -93,7 +92,7 @@ pyyaml = "6.0.1"
 conda-pack = "0.7.1"
 flake8 = "7.0.0"
 flake8-black = "0.3.6"
-mypy = "1.5.1"
+mypy = "1.8.0"
 mypy-protobuf = "3.1.0"
 types-protobuf = "4.24.0.20240129"
 types-orjson = "3.6.2"
@@ -106,7 +105,7 @@ filelock = "^3.13.1"
 
 [tool.poetry.group.docker.dependencies]
 protobuf = "3.20.3"
-tensorflow = "~2.12.0"
+tensorflow = "<2.15"
 
 [tool.poetry.group.all-runtimes]
 optional = true
@@ -129,13 +128,13 @@ requests-mock = "~1.11.0"
 types-requests = "~2.28.11.5"
 
 ## Dev dependencies from Alibi-Detect
-tensorflow = "~2.12.0"
+tensorflow = "<2.15"
 
 ## Dev dependencies from MLflow
-torch = "2.0.0"
-pytorch-lightning = "2.1.3"
+torch = "2.2.1"
+pytorch-lightning = "2.2.0.post0"
 torchmetrics = "1.3.1"
-torchvision = "0.15.1"
+torchvision = "0.17.1"
 mlflow = "2.10.2"
 
 ## Dev dependencies from HuggingFace

diff --git a/runtimes/alibi-detect/poetry.lock b/runtimes/alibi-detect/poetry.lock
diff --git a/runtimes/alibi-detect/pyproject.toml b/runtimes/alibi-detect/pyproject.toml
@@ -8,13 +8,14 @@ readme = "README.md"
 packages = [{include = "mlserver_alibi_detect"}]
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<3.12"
+python = ">=3.9,<3.12"
 mlserver = "*"
 alibi-detect = {extras = ["tensorflow"], version = "*"}
+pydantic = "<2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}
-tensorflow = ">=2.12,<2.14"
+tensorflow = "<2.15"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/runtimes/alibi-explain/poetry.lock b/runtimes/alibi-explain/poetry.lock
diff --git a/runtimes/alibi-explain/pyproject.toml b/runtimes/alibi-explain/pyproject.toml
@@ -8,20 +8,21 @@ readme = "README.md"
 packages = [{include = "mlserver_alibi_explain"}]
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<3.12"
+python = ">=3.9,<3.12"
 mlserver = "*"
 mlserver_sklearn = "*"
 mlserver_xgboost = "*"
 mlserver_lightgbm = "*"
 orjson = "*"
 alibi = {extras = ["shap", "tensorflow"], version = "*"}
+pydantic = "<2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}
 mlserver_sklearn = {path = "../../runtimes/sklearn", develop = true}
 mlserver_xgboost = {path = "../../runtimes/xgboost", develop = true}
 mlserver_lightgbm = {path = "../../runtimes/lightgbm", develop = true}
-tensorflow = "~2.12.0"
+tensorflow = "<2.15"
 requests-mock = "~1.10.0"
 types-requests = "~2.28.11.5"
 

diff --git a/runtimes/catboost/poetry.lock b/runtimes/catboost/poetry.lock
diff --git a/runtimes/catboost/pyproject.toml b/runtimes/catboost/pyproject.toml
@@ -9,10 +9,11 @@ readme = "README.md"
 packages = [{include = "mlserver_catboost"}]
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<3.12"
+python = ">=3.9,<3.12"
 catboost = "*"
 joblib = "*"
 mlserver = "*"
+pydantic = "<2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}

diff --git a/runtimes/huggingface/poetry.lock b/runtimes/huggingface/poetry.lock
diff --git a/runtimes/huggingface/pyproject.toml b/runtimes/huggingface/pyproject.toml
@@ -8,17 +8,18 @@ readme = "README.md"
 packages = [{include = "mlserver_huggingface"}]
 
 [tool.poetry.dependencies]
-python = "^3.8.1,<3.12"
+python = ">=3.9,<3.12"
 mlserver = "*"
 tensorflow = "*"
 pillow = "*"
 optimum = {extras = ["onnxruntime"], version = ">=1.4,<2.0"}
+pydantic = "<2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}
-torch = "2.1.2"
+torch = "2.2.1"
 transformers = ">=4.30,<5.0"
-tensorflow = ">=2.12,<2.14"
+tensorflow = ">=2.12,<2.15"
 
 [build-system]
 requires = ["poetry-core"]