format with ruff

libs/infinity_emb/infinity_emb/engine.py

@@ -2,10 +2,9 @@
 # Copyright (c) 2023-now michaelfeilfeil
 
 from asyncio import Semaphore
-from typing import Iterable, Iterator, Optional, Set, Union, List
+from typing import Iterable, Iterator, List, Optional, Set, Union
 
 from infinity_emb.args import EngineArgs
-from infinity_emb.primitives import ImageClassType
 
 # prometheus
 from infinity_emb.inference import (
@@ -16,6 +15,7 @@
 from infinity_emb.primitives import (
     ClassifyReturnType,
     EmbeddingReturnType,
+    ImageClassType,
     ModelCapabilites,
 )
 

libs/infinity_emb/infinity_emb/inference/batch_handler.py

@@ -9,7 +9,7 @@
 import time
 from concurrent.futures import ThreadPoolExecutor
 from queue import Queue
-from typing import Any, Sequence, Set, Union, List
+from typing import Any, List, Sequence, Set, Union
 
 import numpy as np
 
@@ -23,14 +23,14 @@
     ClassifyReturnType,
     EmbeddingReturnType,
     EmbeddingSingle,
+    ImageClassType,
     ModelCapabilites,
     ModelNotDeployedError,
     OverloadStatus,
     PredictSingle,
     PrioritizedQueueItem,
     ReRankSingle,
     get_inner_item,
-    ImageClassType,
 )
 from infinity_emb.transformer.abstract import BaseTransformer
 from infinity_emb.transformer.utils import get_lengths_with_tokenize

libs/infinity_emb/infinity_emb/primitives.py

@@ -19,20 +19,19 @@
 from functools import lru_cache
 from typing import (
     TYPE_CHECKING,
+    Any,
     Generic,
     Literal,
     Optional,
     Type,
     TypedDict,
     TypeVar,
     Union,
-    Any,
 )
 
 import numpy as np
 import numpy.typing as npt
 
-
 EmptyImageClassType: Any = Any
 if TYPE_CHECKING:
     try:

libs/infinity_emb/infinity_emb/sync_engine.py

@@ -11,9 +11,9 @@
     Awaitable,
     Callable,
     Iterator,
+    List,
     TypeVar,
     Union,
-    List,
 )
 
 from infinity_emb.engine import AsyncEmbeddingEngine, AsyncEngineArray, EngineArgs

libs/infinity_emb/infinity_emb/transformer/vision/utils.py

@@ -1,9 +1,10 @@
 # SPDX-License-Identifier: MIT
 # Copyright (c) 2023-now michaelfeil
 
+from typing import List, Union
+
 from infinity_emb._optional_imports import CHECK_PIL, CHECK_REQUESTS
-from infinity_emb.primitives import ImageCorruption, ImageSingle, ImageClassType
-from typing import Union, List
+from infinity_emb.primitives import ImageClassType, ImageCorruption, ImageSingle
 
 if CHECK_PIL.is_available:
     from PIL import Image  # type: ignore

libs/infinity_emb/tests/unit_test/test_engine.py

@@ -4,10 +4,10 @@
 
 import numpy as np
 import pytest
+import requests
 import torch
-from sentence_transformers import CrossEncoder  # type: ignore[import-untyped]
 from PIL import Image
-import requests
+from sentence_transformers import CrossEncoder  # type: ignore[import-untyped]
 
 from infinity_emb import AsyncEmbeddingEngine, AsyncEngineArray, EngineArgs
 from infinity_emb.primitives import (