From 3e3c1df6ba6c2ea04aeff1397c1beac65d2238ca Mon Sep 17 00:00:00 2001
From: Michael The <michael.the@teampicnic.com>
Date: Fri, 19 Jul 2024 12:52:06 +0200
Subject: [PATCH] small fix

---
 Makefile          | 4 ++--
 mypy.ini          | 3 +++
 src/sqlfmt/api.py | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/Makefile b/Makefile
index 555bf8b0..acb16c47 100644
--- a/Makefile
+++ b/Makefile
@@ -4,7 +4,7 @@ check:
 	black .
 	pytest
 	flake8 .
-	mypy --disallow-any-unimported
+	mypy .
 
 .PHONY: unit
 unit:
@@ -15,7 +15,7 @@ lint:
 	isort .
 	black .
 	flake8 .
-	mypy --disallow-any-unimported
+	mypy .
 
 
 .PHONY: profiling
diff --git a/mypy.ini b/mypy.ini
index 2c1c7745..c4bf7b41 100755
--- a/mypy.ini
+++ b/mypy.ini
@@ -28,3 +28,6 @@ warn_unused_configs=True
 no_implicit_reexport=True
 strict_equality=True
 extra_checks=True
+
+[mypy-mistletoe.*]
+ignore_missing_imports=True
diff --git a/src/sqlfmt/api.py b/src/sqlfmt/api.py
index 759c61ee..b7268a6e 100755
--- a/src/sqlfmt/api.py
+++ b/src/sqlfmt/api.py
@@ -86,7 +86,7 @@ def format_sql_code_blocks(token: BlockToken) -> None:
     with MarkdownRenderer() as renderer:
         doc = Document(source_string)
         format_sql_code_blocks(doc)
-        formatted_markdown_string = renderer.render(doc)
+        formatted_markdown_string: str = renderer.render(doc)
 
     return formatted_markdown_string