Skip to content
Snippets Groups Projects
Commit 1e7a1e6f authored by Markus Holzer's avatar Markus Holzer
Browse files

Remove gmpy workaround

parent cb9cccfb
Branches
Tags
No related merge requests found
"""Module to generate stencil kernels in C or CUDA using sympy expressions and call them as Python functions""" """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 fd
from . import stencil as stencil from . import stencil as stencil
from .assignment import Assignment, assignment_from_stencil from .assignment import Assignment, assignment_from_stencil
......
...@@ -4,6 +4,5 @@ import sympy as sp ...@@ -4,6 +4,5 @@ import sympy as sp
import pystencils as ps import pystencils as ps
import pystencils.jupyter import pystencils.jupyter
import pystencils.plot as plt import pystencils.plot as plt
import pystencils.sympy_gmpy_bug_workaround
__all__ = ['sp', 'np', 'ps', 'plt'] __all__ = ['sp', 'np', 'ps', 'plt']
# 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__ = []
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment