Commit 856b8226 authored by Martin Bauer's avatar Martin Bauer
Browse files

sympy 1.1 compatibility

parent bee3bfdf
import sympy as sp
from sympy.utilities.codegen import CCodePrinter
try:
from sympy.utilities.codegen import CCodePrinter
except ImportError:
from sympy.printing.ccode import C99CodePrinter as CCodePrinter
from pystencils.astnodes import Node
from pystencils.types import createType, PointerType
......
......@@ -477,7 +477,7 @@ def typeAllEquations(eqs, typeForSymbol):
typedEquations = []
for eq in eqs:
if isinstance(eq, sp.Eq):
if isinstance(eq, sp.Eq) or isinstance(eq, ast.SympyAssignment):
newLhs = processLhs(eq.lhs)
newRhs = processRhs(eq.rhs)
typedEquations.append(ast.SympyAssignment(newLhs, newRhs))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment