Skip to content
Snippets Groups Projects

[Fix] Update for Docker Images

Closed Markus Holzer requested to merge holzer/pystencils:SmallFixes into master
@@ -188,13 +188,15 @@ def get_type_of_expression(expr,
# Fix for sympy versions from 1.9
sympy_version = sp.__version__.split('.')
if int(sympy_version[0]) * 100 + int(sympy_version[1]) >= 109:
sympy_version_int = int(sympy_version[0]) * 100 + int(sympy_version[1])
if sympy_version_int >= 109:
# __setstate__ would bypass the contructor, so we remove it
sp.Number.__getstate__ = sp.Basic.__getstate__
del sp.Basic.__getstate__
if sys.version_info.major * 100 + sys.version_info.minor >= 311:
if sympy_version_int >= 111:
del sp.Basic.__setstate__
del sp.Symbol.__setstate__
del sp.Symbol.__setstate__
else:
sp.Number.__getstate__ = sp.Basic.__getstate__
del sp.Basic.__getstate__
# __reduce_ex__ would strip kwargs, so we override it
def basic_reduce_ex(self, protocol):
@@ -207,8 +209,6 @@ if int(sympy_version[0]) * 100 + int(sympy_version[1]) >= 109:
else:
state = None
return partial(type(self), **kwargs), args, state
sp.Number.__reduce_ex__ = sp.Basic.__reduce_ex__
sp.Basic.__reduce_ex__ = basic_reduce_ex