Pin numpy/transformers to prevent runtime error

dottxt-ai · Nov 12, 2024 · 8565981 · 8565981
1 parent c7af981
commit 8565981
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,9 +43,11 @@ test = [
     "beartype<0.16.0",
     "huggingface_hub",
     "torch",
-    "numpy",
+    # transformers 4.46.1/2 fail with RuntimeError:
+    # - numpy.core is deprecated and has been renamed to numpy._core
+    "numpy==2.1.1",
     "scipy",
-    "transformers",
+    "transformers==4.45.2",
     "datasets",
     "pillow",
     "asv",