diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py index 62875a5a115..b0ebcf77673 100644 --- a/tests/local_testing/test_custom_callback_input.py +++ b/tests/local_testing/test_custom_callback_input.py @@ -1339,7 +1339,7 @@ def test_standard_logging_payload_audio(turn_off_message_logging, stream): continue time.sleep(2) - mock_client.assert_called_once() + mock_client.assert_called() print( f"mock_client_post.call_args: {mock_client.call_args.kwargs['kwargs'].keys()}"