Bugfix/torch install on macosx x86 64 (#143)

- Added restriction to `pytorch` version for MacOSX + x86_64 that allows to install it on such platforms
MobileTeleSystems · May 18, 2024 · 980afd6 · 980afd6
1 parent b9ca1a9
commit 980afd6
Show file tree

Hide file tree

Showing 3 changed files with 105 additions and 9 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -11,8 +11,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 - Extended Theory&Practice RecSys baselines tutorial ([#139](https://github.com/MobileTeleSystems/RecTools/pull/139))
 
-### Changed
-- Used the latest version of `lightfm` that allows to install it using `poetry 1.5.0+` ([#141](https://github.com/MobileTeleSystems/RecTools/pull/141))
+### Fixed
+- Used the latest version of `lightfm` that allows to install it using `poetry>=1.5.0` ([#141](https://github.com/MobileTeleSystems/RecTools/pull/141))
+- Added restriction to `pytorch` version for MacOSX + x86_64 that allows to install it on such platforms ([#142](https://github.com/MobileTeleSystems/RecTools/pull/142))
 
 
 ## [0.6.0] - 13.05.2024

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -76,7 +76,11 @@ lightfm = {git = "https://github.com/lyst/lightfm", rev = "0c9c31e", python = "<
 
 nmslib = {version = "^2.0.4", python = "<3.11", optional = true}
 
-torch = {version = ">=1.6.0, <3.0.0", optional = true}
+# The latest torch version available for MacOSX + x86_64 is 2.2.2
+torch = [
+    {version = ">=1.6.0, <2.3.0", markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", optional = true},
+    {version = ">=1.6.0, <3.0.0", optional = true}
+]
 pytorch-lightning = {version = ">=1.6.0, <3.0.0", optional = true}
 
 ipywidgets = {version = ">=7.7,<8.2", optional = true}