diff --git a/pymbolic/interop/sympy.py b/pymbolic/interop/sympy.py index 76ef76c56f9e2436a5261d45a893508cef5dbf4b..86f50dc66929c658fc4de1569d7c4c10f2319bc4 100644 --- a/pymbolic/interop/sympy.py +++ b/pymbolic/interop/sympy.py @@ -109,7 +109,7 @@ class PymbolicToSympyMapper(PymbolicToSympyLikeMapper): *[self.sym.Symbol(v) for v in expr.variables]) def map_subscript(self, expr): - return self.sym.tensor.indexed.Indexed( + return self.sym.Indexed( self.rec(expr.aggregate), *tuple(self.rec(i) for i in expr.index_tuple) ) diff --git a/test/test_sympy.py b/test/test_sympy.py index d8a5c94826d7468e17e234c1d2d685c36afe3aea..3cd55acef24ff512034671c7ee8e5e05e95ad447 100644 --- a/test/test_sympy.py +++ b/test/test_sympy.py @@ -54,7 +54,7 @@ def _test_to_pymbolic(mapper, sym, use_symengine): # indexed accesses if not use_symengine: i, j = sym.symbols("i,j") - assert mapper(sym.tensor.indexed.Indexed(x, i, j)) == x_[i_, j_] + assert mapper(sym.Indexed(x, i, j)) == x_[i_, j_] # constants import math @@ -100,7 +100,7 @@ def _test_from_pymbolic(mapper, sym, use_symengine): assert mapper(x_[0]) == sym.Symbol("x_0") else: i, j = sym.symbols("i,j") - assert mapper(x_[i_, j_]) == sym.tensor.indexed.Indexed(x, i, j) + assert mapper(x_[i_, j_]) == sym.Indexed(x, i, j) assert mapper(prim.Variable("f")(x_)) == sym.Function("f")(x)