diff --git a/src/mapper/dependency.py b/src/mapper/dependency.py index 389ddda36eebd8bc2dbab648dc1761e6815fafd9..aea10c4f3ac0463baa1c46f26f96298727feaf04 100644 --- a/src/mapper/dependency.py +++ b/src/mapper/dependency.py @@ -44,7 +44,10 @@ class DependencyMapper(CombineMapper): return set([expr]) def map_call(self, expr): - if self.include_calls: + if self.include_calls == "descend_args": + return self.combine( + [self.rec(child) for child in expr.parameters]) + elif self.include_calls: return set([expr]) else: return CombineMapper.map_call(self, expr)