From 2ae4a28e6fc59ff57b121bc259a57a1ba3221183 Mon Sep 17 00:00:00 2001
From: Danny McCormick <dannymccormick@google.com>
Date: Mon, 26 Feb 2024 18:18:14 -0500
Subject: [PATCH] Force downgrade mpmath to avoid test breakages (#30418)

---
 sdks/python/tox.ini | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index ca35c383eea1..917eb0bd522c 100644
--- 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'
\ No newline at end of file
+  bash {toxinidir}/scripts/run_pytest.sh {envname} 'apache_beam/ml/transforms/embeddings'