From e4b038faefc1890fea4aadcd5552fc297260acb4 Mon Sep 17 00:00:00 2001 From: KotlinIsland Date: Thu, 29 Apr 2021 16:17:32 +1000 Subject: [PATCH] use type_comments with ast.parse --- src/black/__init__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/black/__init__.py b/src/black/__init__.py index 1c69cc41cdc..5464fb448aa 100644 --- a/src/black/__init__.py +++ b/src/black/__init__.py @@ -6416,7 +6416,12 @@ def parse_ast(src: str) -> Union[ast.AST, ast3.AST, ast27.AST]: # TODO: support Python 4+ ;) for minor_version in range(sys.version_info[1], 4, -1): try: - return ast.parse(src, filename, feature_version=(3, minor_version)) + return ast.parse( + src, + filename, + feature_version=(3, minor_version), + type_comments=True, + ) except SyntaxError: continue else: