def METHOD_NAME(self, input_ast):
    """Updates an abstract syntax tree (AST).
    So that it is compatible with version 2.0 and later of the SageMaker Python SDK.
    Args:
        input_ast (ast.Module): AST to be updated for use with
            the Python SDK version 2.0 and later.
    Returns:
        ast.Module: Updated AST that is compatible with the Python SDK version 2.0 and later.
    """
    return ASTTransformer().visit(input_ast)