diff --git a/tests/llm_translation/test_crusoe.py b/tests/llm_translation/test_crusoe.py index 064f4f053bc..66854b2869f 100644 --- a/tests/llm_translation/test_crusoe.py +++ b/tests/llm_translation/test_crusoe.py @@ -6,7 +6,7 @@ from unittest import mock import litellm -CRUSOE_API_BASE = "https://managed-inference-api-proxy.crusoecloud.com/v1/" +CRUSOE_API_BASE = "https://managed-inference-api-proxy.crusoecloud.com/v1" def test_crusoe_json_registry(): diff --git a/tests/test_litellm/llms/crusoe/test_crusoe.py b/tests/test_litellm/llms/crusoe/test_crusoe.py index 899dab256e3..485c00b357c 100644 --- a/tests/test_litellm/llms/crusoe/test_crusoe.py +++ b/tests/test_litellm/llms/crusoe/test_crusoe.py @@ -1,9 +1,7 @@ import os -import sys - from unittest.mock import patch -CRUSOE_API_BASE = "https://managed-inference-api-proxy.crusoecloud.com/v1/" +CRUSOE_API_BASE = "https://managed-inference-api-proxy.crusoecloud.com/v1" def test_crusoe_json_registry():