def METHOD_NAME():
    fmt_args = ", ".join(
        itertools.chain(
            (repr(pytree.tree_map(_fmt, a)) for a in orig_f_args),
            (
                f"{k}={pytree.tree_map(_fmt, v)}"
                for k, v in orig_f_kwargs.items()
            ),
        )
    )
    return f"{op}({fmt_args})"