From 1e7a1e6f2f8c1aec9673153d4b09b012b06a243b Mon Sep 17 00:00:00 2001
From: markus holzer <markus.holzer@fau.de>
Date: Sat, 26 Jun 2021 09:21:35 +0200
Subject: [PATCH] Remove gmpy workaround
---
pystencils/__init__.py | 1 -
pystencils/session.py | 1 -
pystencils/sympy_gmpy_bug_workaround.py | 18 ------------------
3 files changed, 20 deletions(-)
delete mode 100644 pystencils/sympy_gmpy_bug_workaround.py
diff --git a/pystencils/__init__.py b/pystencils/__init__.py
index 8b616f441..f58f970ea 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 51d2ef627..f7bf2cf2c 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 79bfc983e..000000000
--- 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__ = []
--
GitLab