fix styling

Author: philschmid

src/huggingface_inference_toolkit/diffusers_utils.py

@@ -4,7 +4,6 @@
 
 from huggingface_inference_toolkit.logging import logger
 
-
 _diffusers = importlib.util.find_spec("diffusers") is not None
 
 

src/huggingface_inference_toolkit/handler.py

@@ -7,8 +7,6 @@
     get_pipeline,
 )
 
-from huggingface_inference_toolkit.logging import logger
-
 
 class HuggingFaceHandler:
     """

src/huggingface_inference_toolkit/optimum_utils.py

@@ -1,7 +1,8 @@
 import importlib.util
-from huggingface_inference_toolkit.logging import logger
 import os
 
+from huggingface_inference_toolkit.logging import logger
+
 _optimum_neuron = False
 if importlib.util.find_spec("optimum") is not None:
     if importlib.util.find_spec("optimum.neuron") is not None:

src/huggingface_inference_toolkit/utils.py

@@ -13,6 +13,7 @@
     get_diffusers_pipeline,
     is_diffusers_available,
 )
+from huggingface_inference_toolkit.logging import logger
 from huggingface_inference_toolkit.optimum_utils import (
     get_optimum_neuron_pipeline,
     is_optimum_neuron_available,
@@ -21,8 +22,6 @@
     get_sentence_transformers_pipeline,
     is_sentence_transformers_available,
 )
-from huggingface_inference_toolkit.logging import logger
-
 
 if is_tf_available():
     import tensorflow as tf

src/huggingface_inference_toolkit/vertex_ai_utils.py

@@ -1,11 +1,9 @@
-import logging
 import re
 from pathlib import Path
 from typing import Union
 
 from huggingface_inference_toolkit.logging import logger
 
-
 GCS_URI_PREFIX = "gs://"
 
 

src/huggingface_inference_toolkit/webservice_starlette.py

@@ -19,14 +19,14 @@
 from huggingface_inference_toolkit.handler import (
     get_inference_handler_either_custom_or_default_handler,
 )
+from huggingface_inference_toolkit.logging import logger
 from huggingface_inference_toolkit.serialization.base import ContentType
 from huggingface_inference_toolkit.serialization.json_utils import Jsoner
 from huggingface_inference_toolkit.utils import (
     _load_repository_from_hf,
     convert_params_to_int_or_bool,
 )
 from huggingface_inference_toolkit.vertex_ai_utils import _load_repository_from_gcs
-from huggingface_inference_toolkit.logging import logger
 
 
 async def prepare_model_artifacts():

tests/integ/conftest.py

@@ -3,7 +3,6 @@
 import random
 import socket
 import time
-import uuid
 
 import docker
 import pytest