From 9fe8507906714c01f64f953ecf868dfc6aa05515 Mon Sep 17 00:00:00 2001 From: Hieu Lam Date: Mon, 10 Mar 2025 10:31:21 +0700 Subject: [PATCH] fix: reformatted to unify syntax --- .../markitdown/src/markitdown/converters/_markdownify.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/packages/markitdown/src/markitdown/converters/_markdownify.py b/packages/markitdown/src/markitdown/converters/_markdownify.py index b641fbb..cd9d08a 100644 --- a/packages/markitdown/src/markitdown/converters/_markdownify.py +++ b/packages/markitdown/src/markitdown/converters/_markdownify.py @@ -7,6 +7,7 @@ from magika import Magika magika = Magika() + class _CustomMarkdownify(markdownify.MarkdownConverter): """ A custom version of markdownify's MarkdownConverter. Changes include: @@ -26,7 +27,10 @@ class _CustomMarkdownify(markdownify.MarkdownConverter): if not extracted_code_snippet: return "" result = magika.identify_bytes(extracted_code_snippet.encode()) - if result.status == "ok" and result.prediction.output.group in ["text", "code"]: + if result.status == "ok" and result.prediction.output.group in [ + "text", + "code", + ]: language = result.prediction.output.label return language return ""