SeldonIO · adriangonz · Aug 31, 2023 · Aug 31, 2023 · Aug 31, 2023 · Aug 31, 2023
diff --git a/Dockerfile b/Dockerfile
@@ -1,7 +1,7 @@
 FROM python:3.10-slim AS wheel-builder
 SHELL ["/bin/bash", "-l", "-c"]
 
-ARG POETRY_VERSION="1.4.2"
+ARG POETRY_VERSION="1.6.1"
 
 COPY ./hack/build-wheels.sh ./hack/build-wheels.sh
 COPY ./mlserver ./mlserver

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -131,12 +131,12 @@ tensorflow = "~2.12.0"
 ## Dev dependencies from MLflow
 torch = "2.0.0"
 pytorch-lightning = "2.0.7"
-torchmetrics = "0.11.4"
+torchmetrics = "1.1.0"
 torchvision = "0.15.1"
-mlflow = "2.5.0"
+mlflow = "2.6.0"
 
 ## Dev dependencies from HuggingFace
-transformers = "~4.30.0"
+transformers = ">=4.30,<4.32"
 
 [tool.poetry.group.docs]
 optional = true

diff --git a/runtimes/alibi-detect/poetry.lock b/runtimes/alibi-detect/poetry.lock
diff --git a/runtimes/alibi-explain/poetry.lock b/runtimes/alibi-explain/poetry.lock
diff --git a/runtimes/huggingface/poetry.lock b/runtimes/huggingface/poetry.lock
diff --git a/runtimes/lightgbm/mlserver_lightgbm/lightgbm.py b/runtimes/lightgbm/mlserver_lightgbm/lightgbm.py
@@ -30,5 +30,9 @@ async def predict(self, payload: types.InferenceRequest) -> types.InferenceRespo
         return types.InferenceResponse(
             model_name=self.name,
             model_version=self.version,
-            outputs=[NumpyCodec.encode_output(name="predict", payload=prediction)],
+            outputs=[
+                NumpyCodec.encode_output(
+                    name="predict", payload=prediction  # type: ignore
+                )
+            ],
         )
diff --git a/runtimes/lightgbm/poetry.lock b/runtimes/lightgbm/poetry.lock
diff --git a/runtimes/mlflow/poetry.lock b/runtimes/mlflow/poetry.lock
diff --git a/runtimes/mlflow/pyproject.toml b/runtimes/mlflow/pyproject.toml
@@ -18,7 +18,7 @@ torch = "2.0.0"
 pytorch-lightning = "2.0.7"
 torchmetrics = "1.1.0"
 torchvision = "0.15.1"
-mlflow = "2.4.1"
+mlflow = "2.6.0"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/runtimes/mllib/poetry.lock b/runtimes/mllib/poetry.lock
diff --git a/runtimes/sklearn/poetry.lock b/runtimes/sklearn/poetry.lock
diff --git a/runtimes/xgboost/poetry.lock b/runtimes/xgboost/poetry.lock