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,
 # )

tests/integ/conftest.py

@@ -7,9 +7,9 @@
 import docker
 import pytest
 import tenacity
+from huggingface_inference_toolkit.utils import _load_repository_from_hf
 from transformers.testing_utils import _run_slow_tests
 
-from huggingface_inference_toolkit.utils import _load_repository_from_hf
 from tests.integ.config import task2model
 
 HF_HUB_CACHE = os.environ.get("HF_HUB_CACHE", "/home/ubuntu/.cache/huggingface/hub")

tests/integ/helpers.py

@@ -8,9 +8,9 @@
 import pytest
 import requests
 from docker import DockerClient
+from huggingface_inference_toolkit.utils import _load_repository_from_hf
 from transformers.testing_utils import _run_slow_tests, require_tf, require_torch
 
-from huggingface_inference_toolkit.utils import _load_repository_from_hf
 from tests.integ.config import task2input, task2model, task2output, task2validation
 
 IS_GPU = _run_slow_tests

tests/integ/test_pytorch_local_inf2.py

@@ -1,7 +1,7 @@
 import pytest
+from huggingface_inference_toolkit.optimum_utils import is_optimum_neuron_available
 from transformers.testing_utils import require_torch
 
-from huggingface_inference_toolkit.optimum_utils import is_optimum_neuron_available
 from tests.integ.helpers import verify_task
 
 require_inferentia = pytest.mark.skipif(

tests/unit/test_diffusers.py

@@ -1,11 +1,10 @@
 import logging
 import tempfile
 
-from PIL import Image
-from transformers.testing_utils import require_torch, slow
-
 from huggingface_inference_toolkit.diffusers_utils import IEAutoPipelineForText2Image
 from huggingface_inference_toolkit.utils import _load_repository_from_hf, get_pipeline
+from PIL import Image
+from transformers.testing_utils import require_torch, slow
 
 logging.basicConfig(level="DEBUG")
 

tests/unit/test_handler.py

@@ -2,8 +2,6 @@
 from typing import Dict
 
 import pytest
-from transformers.testing_utils import require_tf, require_torch
-
 from huggingface_inference_toolkit.handler import (
     HuggingFaceHandler,
     get_inference_handler_either_custom_or_default_handler,
@@ -12,6 +10,7 @@
     _is_gpu_available,
     _load_repository_from_hf,
 )
+from transformers.testing_utils import require_tf, require_torch
 
 TASK = "text-classification"
 MODEL = "hf-internal-testing/tiny-random-distilbert"

tests/unit/test_optimum_utils.py

@@ -2,14 +2,13 @@
 import tempfile
 
 import pytest
-from transformers.testing_utils import require_torch
-
 from huggingface_inference_toolkit.optimum_utils import (
     get_input_shapes,
     get_optimum_neuron_pipeline,
     is_optimum_neuron_available,
 )
 from huggingface_inference_toolkit.utils import _load_repository_from_hf
+from transformers.testing_utils import require_torch
 
 require_inferentia = pytest.mark.skipif(
     not is_optimum_neuron_available(),