void (SyntaxTreeBuilder builder) { LOG.assertTrue(builder.getTokenType() == PyTokenTypes.RETURN_KEYWORD); final SyntaxTreeBuilder.Marker returnStatement = builder.mark(); builder.advanceLexer(); if (builder.getTokenType() != null && !getEndOfStatementsTokens().contains(builder.getTokenType())) { getExpressionParser().parseExpression(); } checkEndOfStatement(); returnStatement.done(PyElementTypes.RETURN_STATEMENT); }