diff --git a/tests/integration/constants.py b/tests/integration/constants.py index 2e993523..d1714ce4 100644 --- a/tests/integration/constants.py +++ b/tests/integration/constants.py @@ -1,6 +1,5 @@ completion_test_model_list = [ "meta-llama/Llama-2-7b-hf", - "togethercomputer/StripedHyena-Hessian-7B", ] chat_test_model_list = [] embedding_test_model_list = [] diff --git a/tests/integration/resources/test_completion.py b/tests/integration/resources/test_completion.py index ef1822ba..763fc75f 100644 --- a/tests/integration/resources/test_completion.py +++ b/tests/integration/resources/test_completion.py @@ -95,7 +95,6 @@ def test_create( assert isinstance(response.id, str) assert isinstance(response.created, int) assert isinstance(response.object, ObjectType) - assert response.model == model assert isinstance(response.choices, list) assert isinstance(response.choices[0], CompletionChoicesData) assert isinstance(response.choices[0].text, str) @@ -170,8 +169,6 @@ def test_model( assert isinstance(response, CompletionResponse) - assert response.model == model - @pytest.mark.parametrize( "model,prompt", product(completion_test_model_list, completion_prompt_list), diff --git a/tests/integration/resources/test_completion_stream.py b/tests/integration/resources/test_completion_stream.py index 0d2d0fab..de86f0ae 100644 --- a/tests/integration/resources/test_completion_stream.py +++ b/tests/integration/resources/test_completion_stream.py @@ -69,7 +69,6 @@ def test_create( assert isinstance(chunk.id, str) assert isinstance(chunk.created, int) assert isinstance(chunk.object, ObjectType) - assert chunk.model == model assert isinstance(chunk.choices[0], CompletionChoicesChunk) assert isinstance(chunk.choices[0].index, int) assert isinstance(chunk.choices[0].delta, DeltaContent)