From d0bdafa49c06e6fca8ca7aa697af694af4d31945 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 15 Aug 2024 14:49:21 -0700 Subject: [PATCH] run mock tests for test_completion_sagemaker --- litellm/tests/test_completion.py | 27 --------------------------- 1 file changed, 27 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index cc1b24cde1..28d298d4d0 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -3337,33 +3337,6 @@ def test_customprompt_together_ai(): # test_customprompt_together_ai() -@pytest.mark.skip(reason="AWS Suspended Account") -def test_completion_sagemaker(): - try: - litellm.set_verbose = True - print("testing sagemaker") - response = completion( - model="sagemaker/jumpstart-dft-hf-llm-mistral-7b-ins-20240329-150233", - model_id="huggingface-llm-mistral-7b-instruct-20240329-150233", - messages=messages, - temperature=0.2, - max_tokens=80, - aws_region_name=os.getenv("AWS_REGION_NAME_2"), - aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID_2"), - aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY_2"), - input_cost_per_second=0.000420, - ) - # Add any assertions here to check the response - print(response) - cost = completion_cost(completion_response=response) - print("calculated cost", cost) - assert ( - cost > 0.0 and cost < 1.0 - ) # should never be > $1 for a single completion call - except Exception as e: - pytest.fail(f"Error occurred: {e}") - - # test_completion_sagemaker()