Code Style (#254)

optimum_benchmark/__init__.py

@@ -1,7 +1,7 @@
 from .backends import (
     BackendConfig,
-    IPEXConfig,
     INCConfig,
+    IPEXConfig,
     LlamaCppConfig,
     LLMSwarmConfig,
     ORTConfig,

optimum_benchmark/backends/__init__.py

@@ -1,7 +1,7 @@
 from .config import BackendConfig
+from .ipex.config import IPEXConfig
 from .llama_cpp.config import LlamaCppConfig
 from .llm_swarm.config import LLMSwarmConfig
-from .ipex.config import IPEXConfig
 from .neural_compressor.config import INCConfig
 from .onnxruntime.config import ORTConfig
 from .openvino.config import OVConfig

optimum_benchmark/backends/ipex/backend.py

@@ -1,14 +1,11 @@
-import inspect
 from collections import OrderedDict
 from tempfile import TemporaryDirectory
 from typing import Any, Dict
 
 import torch
 from hydra.utils import get_class
 
-from ...generators.dataset_generator import DatasetGenerator
 from ...import_utils import is_accelerate_available, is_torch_distributed_available
-from ...task_utils import TEXT_GENERATION_TASKS
 from ..base import Backend
 from ..transformers_utils import fast_weights_init
 from .config import IPEXConfig
@@ -33,7 +30,6 @@ def __init__(self, config: IPEXConfig) -> None:
         else:
             raise NotImplementedError(f"IPEXBackend does not support task {self.config.task}")
 
-
     def load(self) -> None:
         self.logger.info("\t+ Creating backend temporary directory")
         self.tmpdir = TemporaryDirectory()

optimum_benchmark/backends/ipex/config.py

@@ -1,11 +1,12 @@
-from dataclasses import dataclass, field
-from typing import Any, Dict, Optional
+from dataclasses import dataclass
+from typing import Optional
 
 from ...import_utils import ipex_version
 from ..config import BackendConfig
 
 TORCH_DTYPES = ["bfloat16", "float16", "float32", "auto"]
 
+
 @dataclass
 class IPEXConfig(BackendConfig):
     name: str = "ipex"
@@ -34,4 +35,3 @@ def __post_init__(self):
 
         if self.torch_dtype is not None and self.torch_dtype not in TORCH_DTYPES:
             raise ValueError(f"`torch_dtype` must be one of {TORCH_DTYPES}. Got {self.torch_dtype} instead.")
-

optimum_benchmark/backends/ipex/utils.py

@@ -7,4 +7,3 @@
     "image-classification": "optimum.intel.IPEXModelForImageClassification",
     "audio-classification": "optimum.intel.IPEXModelForAudioClassification",
 }
-

optimum_benchmark/cli.py

@@ -10,10 +10,10 @@
     Benchmark,
     BenchmarkConfig,
     EnergyStarConfig,
-    IPEXConfig,
     INCConfig,
     InferenceConfig,
     InlineConfig,
+    IPEXConfig,
     LlamaCppConfig,
     LLMSwarmConfig,
     ORTConfig,

optimum_benchmark/import_utils.py

@@ -158,14 +158,17 @@ def onnxruntime_version():
             except importlib.metadata.PackageNotFoundError:
                 return None
 
+
 def openvino_version():
     if _openvino_available:
         return importlib.metadata.version("openvino")
 
+
 def ipex_version():
     if _ipex_available:
         return importlib.metadata.version("intel_extension_for_pytorch")
 
+
 def neural_compressor_version():
     if _neural_compressor_available:
         return importlib.metadata.version("neural_compressor")

setup.py

@@ -69,7 +69,7 @@
     "quality": ["ruff"],
     "testing": ["pytest", "hydra-joblib-launcher"],
     # optimum backends
-    "ipex":[f"optimum[ipex]>={MIN_OPTIMUM_VERSION}"],
+    "ipex": [f"optimum[ipex]>={MIN_OPTIMUM_VERSION}"],
     "openvino": [f"optimum[openvino,nncf]>={MIN_OPTIMUM_VERSION}"],
     "onnxruntime": [f"optimum[onnxruntime]>={MIN_OPTIMUM_VERSION}"],
     "onnxruntime-gpu": [f"optimum[onnxruntime-gpu]>={MIN_OPTIMUM_VERSION}"],