Don't let TokenError bubble up from lib2to3_parse (GH-2343)

[etc/vim.git] / src / black / output.py

diff --git a/src/black/output.py b/src/black/output.py
index c85b253c159c885c2fd2e04f9691fdec279fa736..f030d0a0d08d67f07acc5516749ceb5095913a5a 100644 (file)
--- a/src/black/output.py
+++ b/src/black/output.py
@@ -59,8 +59,8 @@ def diff(a: str, b: str, a_name: str, b_name: str) -> str:
     """Return a unified diff string between strings `a` and `b`."""
     import difflib
 
     """Return a unified diff string between strings `a` and `b`."""
     import difflib
 

-    a_lines = [line for line in a.splitlines(keepends=True)]
-    b_lines = [line for line in b.splitlines(keepends=True)]
+    a_lines = a.splitlines(keepends=True)
+    b_lines = b.splitlines(keepends=True)

     diff_lines = []
     for line in difflib.unified_diff(
         a_lines, b_lines, fromfile=a_name, tofile=b_name, n=5
     diff_lines = []
     for line in difflib.unified_diff(
         a_lines, b_lines, fromfile=a_name, tofile=b_name, n=5