Fix CI/CD macos-huggingface (#1805)

* Constrain pyproject version for macos * Included comment about macos x86 deprecation * Set torch upperbound to 2.4.0
SeldonIO · Jun 14, 2024 · d3c12b6 · d3c12b6
1 parent 3af34a0
commit d3c12b6
Show file tree

Hide file tree

Showing 2 changed files with 128 additions and 32 deletions.
diff --git a/runtimes/huggingface/poetry.lock b/runtimes/huggingface/poetry.lock
diff --git a/runtimes/huggingface/pyproject.toml b/runtimes/huggingface/pyproject.toml
@@ -17,7 +17,12 @@ pydantic = "2.7.1"
 
 [tool.poetry.group.dev.dependencies]
 mlserver = {path = "../..", develop = true}
-torch = "2.3.0"
+# macOS x86 deprecated
+# see: https://github.com/pytorch/pytorch/issues/114602
+torch = [
+    {markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", version = "<2.3.0"},
+    {markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'", version = "<2.4.0"}
+]
 transformers = ">=4.30,<5.0"
 tensorflow = ">=2.12,<2.17"