diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py index 3ed5240587a905c24af3a986232dd0ad06df19ad..ddbe2126cff1d59c062d18187ceaff875ff58022 100644 --- a/pymbolic/mapper/__init__.py +++ b/pymbolic/mapper/__init__.py @@ -748,6 +748,19 @@ class WalkMapper(RecursiveMapper): self.post_visit(expr, *args, **kwargs) + def map_slice(self, expr, *args, **kwargs): + if not self.visit(expr, *args, **kwargs): + return + + if expr.start is not None: + self.rec(expr.start, *args, **kwargs) + if expr.stop is not None: + self.rec(expr.stop, *args, **kwargs) + if expr.step is not None: + self.rec(expr.step, *args, **kwargs) + + self.post_visit(expr, *args, **kwargs) + def visit(self, expr, *args, **kwargs): return True