fix isort

openvinotoolkit · Jun 11, 2024 · fb5b85f · fb5b85f
1 parent b9d80b8
commit fb5b85f
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 9 deletions.
diff --git a/openvino_xai/__init__.py b/openvino_xai/__init__.py
@@ -5,8 +5,8 @@
 """
 
 
+from .api.api import insert_xai
 from .common.parameters import Method, Task
 from .explainer.explainer import Explainer
-from .api.api import insert_xai
 
 __all__ = ["Explainer", "insert_xai", "Method", "Task"]
diff --git a/openvino_xai/api/api.py b/openvino_xai/api/api.py
@@ -1,12 +1,11 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
+import openvino.runtime as ov
+
 from openvino_xai import Task
 from openvino_xai.common.utils import IdentityPreprocessFN, has_xai, logger
 from openvino_xai.inserter.parameters import InsertionParameters
-
-import openvino.runtime as ov
-
 from openvino_xai.methods.factory import WhiteBoxMethodFactory
 
 

diff --git a/openvino_xai/inserter/inserter.py b/openvino_xai/inserter/inserter.py
@@ -4,9 +4,7 @@
 import openvino.runtime as ov
 from openvino.preprocess import PrePostProcessor
 
-from openvino_xai.common.utils import (
-    SALIENCY_MAP_OUTPUT_NAME,
-)
+from openvino_xai.common.utils import SALIENCY_MAP_OUTPUT_NAME
 
 
 def insert_xai_branch_into_model(

diff --git a/tests/unit/common/test_utils.py b/tests/unit/common/test_utils.py
@@ -5,9 +5,9 @@
 
 import openvino.runtime as ov
 
+from openvino_xai.api.api import insert_xai
 from openvino_xai.common.parameters import Task
 from openvino_xai.common.utils import has_xai, retrieve_otx_model
-from openvino_xai.api.api import insert_xai
 from tests.integration.test_classification import DEFAULT_CLS_MODEL
 
 DARA_DIR = Path(".data")

diff --git a/tests/unit/explanation/test_explainer.py b/tests/unit/explanation/test_explainer.py
@@ -7,6 +7,7 @@
 import openvino.runtime as ov
 import pytest
 
+from openvino_xai.api.api import insert_xai
 from openvino_xai.common.parameters import Task
 from openvino_xai.common.utils import retrieve_otx_model
 from openvino_xai.explainer.explainer import Explainer
@@ -16,7 +17,6 @@
     TargetExplainGroup,
 )
 from openvino_xai.explainer.utils import get_postprocess_fn, get_preprocess_fn
-from openvino_xai.api.api import insert_xai
 from openvino_xai.inserter.parameters import ClassificationInsertionParameters
 from tests.unit.explanation.test_explanation_utils import VOC_NAMES