Skip to content

Commit f46b368

Browse files
fix black import issue
1 parent d2a8711 commit f46b368

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

codeflash/code_utils/formatter.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@
1515

1616
def get_diff_output_by_black(filepath: str, unformatted_content: str) -> Optional[str]:
1717
try:
18-
import black
18+
from black import Mode, format_file_contents, output
1919

20-
formatted_content = black.format_file_contents(src_contents=unformatted_content, fast=True, mode=black.Mode())
21-
return black.diff(unformatted_content, formatted_content, a_name=filepath, b_name=filepath)
20+
formatted_content = format_file_contents(src_contents=unformatted_content, fast=True, mode=Mode())
21+
return output.diff(unformatted_content, formatted_content, a_name=filepath, b_name=filepath)
2222
except ImportError:
2323
return None
2424

0 commit comments

Comments
 (0)