From a0aa9dd07dbc5004aeb7c4cc2da26c14b809630b Mon Sep 17 00:00:00 2001
From: Max Balandat <balandat@meta.com>
Date: Sun, 17 Mar 2024 21:26:00 -0700
Subject: [PATCH] Pin mpmath to <=1.3.0

Without this, this can cause breakages with pytorch nightlies, see https://github.com/pytorch/pytorch/issues/120995
---
 setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 4cf95ef0..0fa7de65 100644
--- a/setup.py
+++ b/setup.py
@@ -37,7 +37,12 @@
     pass
 
 # Other requirements
-install_requires += ["scipy", "jaxtyping>=0.2.9", "typeguard~=2.13.3"]
+install_requires += [
+    "scipy",
+    "jaxtyping>=0.2.9",
+    "typeguard~=2.13.3",
+    "mpmath>=0.19,<=1.3",  # avoid incompatibiltiy with torch+sympy with mpmath 1.4
+]
 
 
 # Get version