Mock more e2b none

2024-12-30 15:04:08 -08:00
parent 87cc9d2516
commit e3623d142c
1 changed files with 1 additions and 1 deletions
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -385,7 +385,7 @@ def test_function_return_limit(mock_e2b_api_key_none, client: Union[LocalClient,
    client.delete_agent(agent_id=agent.id)


-def test_function_always_error(client: Union[LocalClient, RESTClient]):
+def test_function_always_error(mock_e2b_api_key_none, client: Union[LocalClient, RESTClient]):
    """Test to see if function that errors works correctly"""

    def always_error():