From c2447633f654db1d91706ab995b6817c70f1d66e Mon Sep 17 00:00:00 2001
From: Nicholas Christensen <njchris2@illinois.edu>
Date: Sun, 2 Oct 2022 18:28:40 -0500
Subject: [PATCH] kwargs on map_if
---
pymbolic/mapper/__init__.py | 4 ++--
pymbolic/mapper/dependency.py | 6 +++---
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py
index d1dc525..dd673c2 100644
--- a/pymbolic/mapper/__init__.py
+++ b/pymbolic/mapper/__init__.py
@@ -398,13 +398,13 @@ class CombineMapper(RecursiveMapper):
def map_common_subexpression(self, expr, *args, **kwargs):
return self.rec(expr.child, *args, **kwargs)
- def map_if_positive(self, expr):
+ def map_if_positive(self, expr, *args, **kwargs):
return self.combine([
self.rec(expr.criterion),
self.rec(expr.then),
self.rec(expr.else_)])
- def map_if(self, expr):
+ def map_if(self, expr, *args, **kwargs):
return self.combine([
self.rec(expr.condition),
self.rec(expr.then),
diff --git a/pymbolic/mapper/dependency.py b/pymbolic/mapper/dependency.py
index 20d502a..0cffd44 100644
--- a/pymbolic/mapper/dependency.py
+++ b/pymbolic/mapper/dependency.py
@@ -85,19 +85,19 @@ class DependencyMapper(CSECachingMapperMixin, Collector):
elif self.include_calls:
return {expr}
else:
- return super().map_call_with_kwargs(expr)
+ return super().map_call_with_kwargs(expr, *args, **kwargs)
def map_lookup(self, expr, *args, **kwargs):
if self.include_lookups:
return {expr}
else:
- return super().map_lookup(expr)
+ return super().map_lookup(expr, *args, **kwargs)
def map_subscript(self, expr, *args, **kwargs):
if self.include_subscripts:
return {expr}
else:
- return super().map_subscript(expr)
+ return super().map_subscript(expr, *args, **kwargs)
def map_common_subexpression_uncached(self, expr):
if self.include_cses:
--
GitLab