Merge pull request #417 from MarcoGorelli/use-comprehension

CLN - use dict comprehension in _format_params
asottile · Apr 3, 2021 · 5495a24 · 5495a24
2 parents 2e32e37 + b468218
commit 5495a24
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pyupgrade/_main.py b/pyupgrade/_main.py
@@ -523,9 +523,7 @@ def _simple_arg(arg: ast.expr) -> bool:
 
 
 def _format_params(call: ast.Call) -> Dict[str, str]:
-    params = {}
-    for i, arg in enumerate(call.args):
-        params[str(i)] = _unparse(arg)
+    params = {str(i): _unparse(arg) for i, arg in enumerate(call.args)}
     for kwd in call.keywords:
         # kwd.arg can't be None here because we exclude starargs
         assert kwd.arg is not None