Only call get_future_imports when needed (#3135)

This commit is contained in:
Yilei "Dolee" Yang 2022-06-23 12:41:05 -07:00 committed by GitHub
parent 6463fb874f
commit fa6caa6ca8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1172,10 +1172,10 @@ def f(
def _format_str_once(src_contents: str, *, mode: Mode) -> str:
src_node = lib2to3_parse(src_contents.lstrip(), mode.target_versions)
dst_contents = []
future_imports = get_future_imports(src_node)
if mode.target_versions:
versions = mode.target_versions
else:
future_imports = get_future_imports(src_node)
versions = detect_target_versions(src_node, future_imports=future_imports)
normalize_fmt_off(src_node, preview=mode.preview)