def map_call(self, expr): if isinstance(expr.function, prim.Variable): name = expr.function.name if name in ["hankel_1", "bessel_j"]: order, arg = expr.parameters return getattr(self.bessel_getter, name)(order, self.rec(arg)) return IdentityMapper.map_call(self, expr)