diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py index d1dc5253d4c162f8e3f5a529c4fa175a21f746bc..dd673c2ead541411523eb8988df8d36c910e1ed1 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 20d502a7ce03455ba4579ea7590512d0691f1c0d..0cffd44c687215c210ffa0271f630da47b8b6b25 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: