Force downgrade mpmath to avoid test breakages (#30418)

apache · Feb 26, 2024 · 2ae4a28 · 2ae4a28
1 parent 6406cfe
commit 2ae4a28
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
@@ -341,7 +341,9 @@ commands =
 
 [testenv:py{38,39,310}-pytorch-200]
 deps =
-  200: torch>=2.0.0,<2.1.0
+  200:
+    torch>=2.0.0,<2.1.0
+    mpmath==1.3.0
 extras = test,gcp
 # Don't set TMPDIR to avoid "AF_UNIX path too long" errors in certain tests.
 setenv =
@@ -453,4 +455,4 @@ commands =
   # Log aiplatform and its dependencies version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"
   # Allow exit code 5 (no tests run) so that we can run this command safely on arbitrary subdirectories.
-  bash {toxinidir}/scripts/run_pytest.sh {envname} 'apache_beam/ml/transforms/embeddings'
+  bash {toxinidir}/scripts/run_pytest.sh {envname} 'apache_beam/ml/transforms/embeddings'