Add back more e2b

2024-12-30 16:33:59 -08:00
parent f04d5df4ae
commit 172d7ac147
1 changed files with 1 additions and 1 deletions
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -347,7 +347,7 @@ def test_send_system_message(client: Union[LocalClient, RESTClient], agent: Agen
    assert send_system_message_response, "Sending message failed"


-def test_function_return_limit(mock_e2b_api_key_none, client: Union[LocalClient, RESTClient]):
+def test_function_return_limit(client: Union[LocalClient, RESTClient]):
    """Test to see if the function return limit works"""

    def big_return():