Use Ellipses in functions with docstrings (#109)

* Ignore unnecessary ... error * Add back ... to functions with docstring
jackmpcollins · Feb 19, 2024 · 1028c8b · 1028c8b
1 parent 64c6514
commit 1028c8b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -69,6 +69,7 @@ ignore = [
    "COM",  # flake8-commas
     "FA",  # flake8-future-annotations
    "INP",  # flake8-no-pep420
+"PIE790",  # flake8-pie: Unnecessary `...` literal
    "SLF",  # flake8-self
    "ARG",  # flake8-unused-arguments
     "TD",  # flake8-todos

diff --git a/tests/test_chatprompt.py b/tests/test_chatprompt.py
@@ -95,6 +95,7 @@ def test_chatprompt_decorator_docstring():
     @chatprompt(UserMessage("This is a user message."))
     def func(one: int) -> str:
         """This is the docstring."""
+        ...
 
     assert isinstance(func, ChatPromptFunction)
     assert getdoc(func) == "This is the docstring."
@@ -127,6 +128,7 @@ async def test_async_chatprompt_decorator_docstring():
     @chatprompt(UserMessage("This is a user message."))
     async def func(one: int) -> str:
         """This is the docstring."""
+        ...
 
     assert isinstance(func, AsyncChatPromptFunction)
     assert getdoc(func) == "This is the docstring."