Skip to content

Commit f4f6d6d

Browse files
committed
make it work with older huggingface sdk
1 parent d8c89a3 commit f4f6d6d

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/integrations/huggingface_hub/test_huggingface_hub.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -403,9 +403,9 @@ def test_chat_completion_streaming(
403403
}
404404
# usage is not available in older versions of the library
405405
if HF_VERSION and HF_VERSION >= (0, 26, 0):
406-
expected_data["gen_ai.usage.input_tokens"] = (183,)
407-
expected_data["gen_ai.usage.output_tokens"] = (14,)
408-
expected_data["gen_ai.usage.total_tokens"] = (197,)
406+
expected_data["gen_ai.usage.input_tokens"] = 183
407+
expected_data["gen_ai.usage.output_tokens"] = 14
408+
expected_data["gen_ai.usage.total_tokens"] = 197
409409

410410
if send_default_pii:
411411
expected_data["gen_ai.request.messages"] = (

0 commit comments

Comments
 (0)