huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 15:52:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clippy.

Browse Source
This commit is contained in:
Nicolas Patry 2025-03-06 16:31:20 +01:00
parent 818c8db29a
commit 9aa71d61fb
No known key found for this signature in database
GPG Key ID: 4242CEF24CB6DBF9
3 changed files with 0 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
integration-tests/conftest.py
View File

@ -39,13 +39,6 @@ from typing import Dict, List, Optional
from aiohttp import ClientConnectorError, ClientOSError, ServerDisconnectedError from aiohttp import ClientConnectorError, ClientOSError, ServerDisconnectedError
from docker.errors import NotFound from docker.errors import NotFound
from syrupy.extensions.json import JSONSnapshotExtension from syrupy.extensions.json import JSONSnapshotExtension
from huggingface_hub.inference._generated.types.chat_completion import (
ChatCompletionStreamOutput,
ChatCompletionOutput,
)
from openai.types.chat.chat_completion_chunk import (
ChatCompletionChunk as OAIChatCompletionChunk,
)
from text_generation import AsyncClient from text_generation import AsyncClient
from text_generation.types import ( from text_generation.types import (
BestOfSequence, BestOfSequence,

1
integration-tests/models/test_chat_stream_options.py
View File

@ -13,4 +13,3 @@ def chat_handle(launcher):
async def chat_client(chat_handle): async def chat_client(chat_handle):
await chat_handle.health(300) await chat_handle.health(300)
return chat_handle.client return chat_handle.client

3
integration-tests/models/test_completion_prompts.py
View File

@ -1,10 +1,7 @@
import pytest import pytest
import requests import requests
import json
from aiohttp import ClientSession
from openai import OpenAI from openai import OpenAI
from huggingface_hub import InferenceClient from huggingface_hub import InferenceClient
from text_generation.types import Completion
@pytest.fixture(scope="module") @pytest.fixture(scope="module")