From 9aa71d61fb217384b4c3a4811a3eb7965ee3fc75 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 6 Mar 2025 16:31:20 +0100 Subject: [PATCH] Clippy. --- integration-tests/conftest.py | 7 ------- integration-tests/models/test_chat_stream_options.py | 1 - integration-tests/models/test_completion_prompts.py | 3 --- 3 files changed, 11 deletions(-) diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py index e7e64072..be73c60a 100644 --- a/integration-tests/conftest.py +++ b/integration-tests/conftest.py @@ -39,13 +39,6 @@ from typing import Dict, List, Optional from aiohttp import ClientConnectorError, ClientOSError, ServerDisconnectedError from docker.errors import NotFound 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.types import ( BestOfSequence, diff --git a/integration-tests/models/test_chat_stream_options.py b/integration-tests/models/test_chat_stream_options.py index 41f4f741..f93cc951 100644 --- a/integration-tests/models/test_chat_stream_options.py +++ b/integration-tests/models/test_chat_stream_options.py @@ -13,4 +13,3 @@ def chat_handle(launcher): async def chat_client(chat_handle): await chat_handle.health(300) return chat_handle.client - diff --git a/integration-tests/models/test_completion_prompts.py b/integration-tests/models/test_completion_prompts.py index de04d85b..90e26286 100644 --- a/integration-tests/models/test_completion_prompts.py +++ b/integration-tests/models/test_completion_prompts.py @@ -1,10 +1,7 @@ import pytest import requests -import json -from aiohttp import ClientSession from openai import OpenAI from huggingface_hub import InferenceClient -from text_generation.types import Completion @pytest.fixture(scope="module")