Fix InferenceEndpointsLLM unit tests after release of

`huggingface_hub==0.24.0`
argilla-io · Jul 19, 2024 · 01a4bad · 01a4bad
1 parent 9d31547
commit 01a4bad
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/unit/llms/huggingface/test_inference_endpoints.py b/tests/unit/llms/huggingface/test_inference_endpoints.py
@@ -171,7 +171,6 @@ async def test_agenerate_with_chat_completion(
                 created=1721045246,
                 id="",
                 model="meta-llama/Meta-Llama-3-70B-Instruct",
-                object="chat.completion",
                 system_fingerprint="2.1.1-dev0-sha-4327210",
                 usage=ChatCompletionOutputUsage(
                     completion_tokens=66, prompt_tokens=18, total_tokens=84
@@ -212,7 +211,6 @@ async def test_agenerate_with_chat_completion_fails(
                 created=1721045246,
                 id="",
                 model="meta-llama/Meta-Llama-3-70B-Instruct",
-                object="chat.completion",
                 system_fingerprint="2.1.1-dev0-sha-4327210",
                 usage=ChatCompletionOutputUsage(
                     completion_tokens=66, prompt_tokens=18, total_tokens=84