diff --git a/pystencils/assignment.py b/pystencils/assignment.py index 45a95d41ecd643618a3d3b6b4a8a9aa0455a2a02..0bf68799491be29886d215d5ff76010c034bd174 100644 --- a/pystencils/assignment.py +++ b/pystencils/assignment.py @@ -49,6 +49,17 @@ else: __str__ = assignment_str _print_Assignment = print_assignment_latex +# Apparently, in SymPy 1.4 Assignment.__hash__ is not implemented. This has been fixed in current master +try: + sympy_version = sp.__version__.split('.') + + if int(sympy_version[0]) <= 1 and int(sympy_version[1]) <= 4: + def hash_fun(self): + return hash((self.lhs, self.rhs)) + Assignment.__hash__ = hash_fun +except Exception: + pass + def assignment_from_stencil(stencil_array, input_field, output_field, normalization_factor=None, order='visual') -> Assignment: