Merge pull request #132 from ArneBinder/torch-macos-dependency

Set specific pytorch version for major platforms (linux, win32, darwin)
ArneBinder · Jul 12, 2024 · 38b9580 · 38b9580
2 parents 7eb987b + bc35e02
commit 38b9580
Show file tree

Hide file tree

Showing 2 changed files with 230 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,12 @@ pyarrow = "^13"
 
 [tool.poetry.group.dev.dependencies]
 pie-modules = ">=0.11.1,<0.13.0"
-torch = {version = "^2.1.0+cpu", source = "pytorch"}
+torch = [
+    # default
+    {version = "^2.1.0+cpu", source = "pytorch", markers = "sys_platform != 'darwin' or platform_machine != 'x86_64'"},
+    # torch >= 2.3.0 is not available for macOS on intel
+    {version = ">=2.1.0,<2.3.0", source = "pypi", markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'"},
+]
 pytest = "^7.4.2"
 pytest-cov = "^4.1.0"
 pre-commit = "^3.4.0"