Changing return of relay.Function or IRModule?

return this super().visit_call(call)