fix tests

togethercomputer · Aug 26, 2024 · ef984a2 · ef984a2
1 parent b248178
commit ef984a2
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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)