(test) fix incorrect test lol

This commit is contained in:
ishaan-jaff 2024-01-19 19:43:41 -08:00
parent 2fdd41ac1e
commit 5bd10e303a

View file

@ -1384,10 +1384,10 @@ def test_completion_sagemaker():
# 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
raise Exception("it worked!")
except Exception as e:
pytest.fail(f"Error occurred: {e}")