From 2edbaa6e7a0cc1ec602ac6686e40f3c0e01b346a Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 21 Jun 2023 21:36:36 -0700 Subject: [PATCH] fixed tests --- tests/test_coder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_coder.py b/tests/test_coder.py index aa9a678b4..3e5ec63f6 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -138,7 +138,7 @@ class TestCoder(unittest.TestCase): ] # Call the send_with_retries method - coder.send_with_retries("model", ["message"]) + coder.send_with_retries("model", ["message"], None) # Assert that print was called once mock_print.assert_called_once() @@ -160,7 +160,7 @@ class TestCoder(unittest.TestCase): ] # Call the send_with_retries method - coder.send_with_retries("model", ["message"]) + coder.send_with_retries("model", ["message"], None) # Assert that print was called once mock_print.assert_called_once()