phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: fix imports

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-28 13:28:01 -07:00
parent 3560f0ef2c
commit 5ad01e59f6
8 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_sagemaker.py
View file

@ -9,6 +9,8 @@ load_dotenv()
import io
import os
from tests.local_testing.test_streaming import streaming_format_tests
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
@ -130,8 +132,6 @@ async def test_completion_sagemaker_messages_api(sync_mode):
@pytest.mark.flaky(retries=3, delay=1)
async def test_completion_sagemaker_stream(sync_mode, model):
try:
from litellm.tests.test_streaming import streaming_format_tests
litellm.set_verbose = False
print("testing sagemaker")
verbose_logger.setLevel(logging.DEBUG)