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