diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py index 4833296e8..d2e73ce6c 100644 --- a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py +++ b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py @@ -89,7 +89,13 @@ def chat_client(): @pytest.fixture def human_chat_message(chat_client): - return HumanChatMessage(id="test", time=0, body="test message", prompt="test message", client=chat_client) + return HumanChatMessage( + id="test", + time=0, + body="test message", + prompt="test message", + client=chat_client, + ) def test_learn_index_permissions(tmp_path):