From ece0626edcd134da9b0a772b1c8311d35b146a7c Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Mon, 2 Sep 2019 11:53:19 -0500
Subject: [PATCH] WalkMapper walks Slice

---
 pymbolic/mapper/__init__.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py
index 3ed5240..ddbe212 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
 
-- 
GitLab