Fix pre-commit

openvinotoolkit · Feb 2, 2024 · 180e18f · 180e18f
1 parent 4520cd8
commit 180e18f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/unit/algo/hooks/test_xai_hooks.py b/tests/unit/algo/hooks/test_xai_hooks.py
@@ -1,11 +1,7 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 import torch
-
 from datumaro import Polygon
-from torch import LongTensor
-from torchvision import tv_tensors
-
 from otx.algo.hooks.recording_forward_hook import (
     ActivationMapHook,
     DetClassProbabilityMapHook,
@@ -15,6 +11,8 @@
 )
 from otx.core.data.entity.base import ImageInfo
 from otx.core.data.entity.instance_segmentation import InstanceSegBatchPredEntity
+from torch import LongTensor
+from torchvision import tv_tensors
 
 
 def test_activationmap() -> None: