fix: type checking

promplate · Nov 11, 2024 · 416571f · 416571f
1 parent 93e8ca3
commit 416571f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/promplate_recipes/context.py b/promplate_recipes/context.py
@@ -8,7 +8,6 @@
 from promplate.prompt.utils import get_builtins
 
 
-@partial(partial, default_box=True)
 class SilentBox(Box):
     def __str__(self):
         return super().__str__() if len(self) else ""
@@ -20,6 +19,9 @@ def __call__(self, *args, **kwargs):
             return ""
 
 
+SilentBox = partial(SilentBox, default_box=True)  # type: ignore
+
+
 class BuiltinsLayer(dict):
     def __getitem__(self, key):
         return get_builtins()[key]

diff --git a/pyproject.toml b/pyproject.toml
@@ -20,6 +20,9 @@ fmt = { composite = ["ruff check --fix --exit-zero", "isort .", "black ."] }
 [tool.black]
 line-length = 120
 
+[tool.pyright]
+typeCheckingMode = "standard"
+
 [build-system]
 requires = ["pdm-backend"]
 build-backend = "pdm.backend"