diff --git a/pystencils/__init__.py b/pystencils/__init__.py index 8b616f4419d736d602f0108e724c782f9b4b74e3..f58f970ea3b216a8233f1386dddbae74c5e0dc04 100644 --- a/pystencils/__init__.py +++ b/pystencils/__init__.py @@ -1,5 +1,4 @@ """Module to generate stencil kernels in C or CUDA using sympy expressions and call them as Python functions""" -from . import sympy_gmpy_bug_workaround # NOQA from . import fd from . import stencil as stencil from .assignment import Assignment, assignment_from_stencil diff --git a/pystencils/session.py b/pystencils/session.py index 51d2ef627282dba4caea1da53c2132c5ee97aca6..f7bf2cf2c8e9d87548c88ff7afa08cc1d24afcac 100644 --- a/pystencils/session.py +++ b/pystencils/session.py @@ -4,6 +4,5 @@ import sympy as sp import pystencils as ps import pystencils.jupyter import pystencils.plot as plt -import pystencils.sympy_gmpy_bug_workaround __all__ = ['sp', 'np', 'ps', 'plt'] diff --git a/pystencils/sympy_gmpy_bug_workaround.py b/pystencils/sympy_gmpy_bug_workaround.py deleted file mode 100644 index 79bfc983ea4464e67a6d4f7f2ded543eb83fdfbc..0000000000000000000000000000000000000000 --- a/pystencils/sympy_gmpy_bug_workaround.py +++ /dev/null @@ -1,18 +0,0 @@ -# Disable gmpy backend until this bug is resolved if joblib serialize -# See https://github.com/sympy/sympy/pull/13530 -import os -import warnings - -os.environ['MPMATH_NOGMPY'] = '1' -try: - import mpmath.libmp - # In case the user has imported sympy first, then pystencils - if mpmath.libmp.BACKEND == 'gmpy': - warnings.warn("You are using the gmpy backend. You might encounter an error 'argument is not an mpz sympy'. " - "This is due to a known bug in sympy/gmpy library. " - "To prevent this, import pystencils first then sympy or set the environment variable " - "MPMATH_NOGMPY=1") -except ImportError: - pass - -__all__ = []