Remove unused imports from reranks (opea-project#751)

* Remove unused imports Signed-off-by: Dan Middleton <[email protected]> * Remove unused imports Signed-off-by: Dan Middleton <[email protected]> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Dan Middleton <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
wangkl2 · Sep 30, 2024 · 405a2fc · 405a2fc
1 parent 9df4b3c
commit 405a2fc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/comps/reranks/mosec/langchain/dependency/server-ipex.py b/comps/reranks/mosec/langchain/dependency/server-ipex.py
@@ -4,17 +4,16 @@
 import json
 import os
 from os import environ
-from typing import Any, Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union
 
 import intel_extension_for_pytorch as ipex
 import numpy as np
 import torch
 from mosec import Server, Worker
-from mosec.mixin import TypedMsgPackMixin
 from msgspec import Struct
 from sentence_transformers import CrossEncoder
 from torch.utils.data import DataLoader
-from tqdm.autonotebook import tqdm, trange
+from tqdm.autonotebook import tqdm
 
 DEFAULT_MODEL = "/home/user/bge-reranker-large"
 

diff --git a/comps/reranks/tei/local_reranking.py b/comps/reranks/tei/local_reranking.py
@@ -5,15 +5,7 @@
 
 from sentence_transformers import CrossEncoder
 
-from comps import (
-    CustomLogger,
-    RerankedDoc,
-    SearchedDoc,
-    ServiceType,
-    TextDoc,
-    opea_microservices,
-    register_microservice,
-)
+from comps import CustomLogger, RerankedDoc, SearchedDoc, ServiceType, opea_microservices, register_microservice
 
 logger = CustomLogger("local_reranking")
 logflag = os.getenv("LOGFLAG", False)