use torch<2.3.0 because mojmir's laptop sucks

owid · Nov 11, 2024 · 69e2935 · 69e2935
1 parent 34c5186
commit 69e2935
Show file tree

Hide file tree

Showing 2 changed files with 654 additions and 50 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -66,7 +66,6 @@ dependencies = [
     "walden",
     "deprecated>=1.2.14",
     "scikit-learn>=1.5.2",
-    "sentence-transformers>=2.2.2"
 ]
 
 [tool.uv.sources]
@@ -128,6 +127,9 @@ wizard = [
     "pyproj>=3.6.1",
     "streamlit-feedback>=0.1.3",
     "statsmodels>=0.14.4",
+    # Starting from PyTorch 2.3.0, the minimum requirement for macOS is macOS 11.0+ ARM64. Some poor people on the team still use it. Jeez...
+    "torch<2.3.0",
+    "sentence-transformers>=2.2.2",
 ]
 
 [project.scripts]