quality check

Signed-off-by: Raphael Glon <[email protected]>

Author: oOraph

src/huggingface_inference_toolkit/heavy_utils.py

@@ -7,7 +7,6 @@
 from typing import Optional, Union
 
 from huggingface_hub import HfApi, login, snapshot_download
-
 from transformers import WhisperForConditionalGeneration, pipeline
 from transformers.file_utils import is_tf_available, is_torch_available
 from transformers.pipelines import Pipeline
@@ -17,15 +16,15 @@
     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,
+)
 from huggingface_inference_toolkit.sentence_transformers_utils import (
     get_sentence_transformers_pipeline,
     is_sentence_transformers_available,
 )
 from huggingface_inference_toolkit.utils import create_artifact_filter
-from huggingface_inference_toolkit.optimum_utils import (
-    get_optimum_neuron_pipeline,
-    is_optimum_neuron_available,
-)
 
 
 def load_repository_from_hf(
@@ -185,4 +184,4 @@ def get_pipeline(
         hf_pipeline.model.config.forced_decoder_ids = hf_pipeline.tokenizer.get_decoder_prompt_ids(
             language="english", task="transcribe"
         )
-    return hf_pipeline  # type: ignore
+    return hf_pipeline  # type: ignore

src/huggingface_inference_toolkit/idle.py

@@ -5,7 +5,6 @@
 import signal
 import time
 
-
 LOG = logging.getLogger(__name__)
 
 LAST_START = None

src/huggingface_inference_toolkit/utils.py

@@ -5,7 +5,6 @@
 from huggingface_inference_toolkit.const import HF_DEFAULT_PIPELINE_NAME, HF_MODULE_NAME
 from huggingface_inference_toolkit.logging import logger
 
-
 _optimum_available = importlib.util.find_spec("optimum") is not None
 
 

src/huggingface_inference_toolkit/webservice_starlette.py

@@ -27,6 +27,7 @@
 from huggingface_inference_toolkit.serialization.base import ContentType
 from huggingface_inference_toolkit.serialization.json_utils import Jsoner
 from huggingface_inference_toolkit.utils import convert_params_to_int_or_bool
+
 #     _load_repository_from_hf,
 #     convert_params_to_int_or_bool,
 # )