From 513f06be46340e58afdebc80998a3cbf2d390c46 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 28 Oct 2024 14:28:49 -0700 Subject: [PATCH] refactor: move retry_exceptions import to top of test file --- tests/basic/test_sendchat.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/basic/test_sendchat.py b/tests/basic/test_sendchat.py index eb00a7f98..c2cabc2ae 100644 --- a/tests/basic/test_sendchat.py +++ b/tests/basic/test_sendchat.py @@ -4,7 +4,7 @@ from unittest.mock import MagicMock, patch import httpx from aider.llm import litellm -from aider.sendchat import simple_send_with_retries +from aider.sendchat import retry_exceptions, simple_send_with_retries class PrintCalled(Exception): @@ -14,9 +14,6 @@ class PrintCalled(Exception): class TestSendChat(unittest.TestCase): def test_retry_exceptions(self): """Test that retry_exceptions() can be called without raising errors""" - # ai! imports to top - from aider.sendchat import retry_exceptions - retry_exceptions() # Should not raise any exceptions @patch("litellm.completion")