Pin numpy/transformers to prevent runtime error

dottxt-ai · Nov 12, 2024 · 10f31fd · 10f31fd
1 parent c7af981
commit 10f31fd
Showing 1 changed file with 5 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",
@@ -80,6 +82,7 @@ filterwarnings = [
     "error",
     "ignore::pydantic.warnings.PydanticDeprecatedSince20",
     "ignore::UserWarning",
+    "ignore::DeprecationWarning",
 ]
 addopts = [
     "--import-mode=importlib"