diff --git a/tests/unit/data/base/__init__.py b/tests/unit/data/base/__init__.py
index 747ae2b123..3f64d0cf5b 100644
--- a/tests/unit/data/base/__init__.py
+++ b/tests/unit/data/base/__init__.py
@@ -8,7 +8,6 @@
 from .test_image import _TestAnomalibImageDatamodule
 from .test_video import _TestAnomalibVideoDatamodule
 
-
 __all__ = [
     "_TestAnomalibDataModule",
     "_TestAnomalibDepthDatamodule",
diff --git a/tests/unit/data/base/test_datamodule.py b/tests/unit/data/base/test_datamodule.py
index e85a7ec520..7c1adc2d98 100644
--- a/tests/unit/data/base/test_datamodule.py
+++ b/tests/unit/data/base/test_datamodule.py
@@ -4,10 +4,11 @@
 # SPDX-License-Identifier: Apache-2.0
 
 
-from anomalib.data import AnomalibDataModule
 import pytest
 from torch.utils.data import DataLoader
 
+from anomalib.data import AnomalibDataModule
+
 
 class _TestAnomalibDataModule:
     """Base test class for ``AnomalibDataModule``.
diff --git a/tests/unit/data/base/test_depth.py b/tests/unit/data/base/test_depth.py
index 9f0e4ab3a2..ecb719a0da 100644
--- a/tests/unit/data/base/test_depth.py
+++ b/tests/unit/data/base/test_depth.py
@@ -3,8 +3,10 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import AnomalibDataModule
 import pytest
+
+from anomalib.data import AnomalibDataModule
+
 from .test_datamodule import _TestAnomalibDataModule
 
 
diff --git a/tests/unit/data/base/test_image.py b/tests/unit/data/base/test_image.py
index 9862054573..abb0496325 100644
--- a/tests/unit/data/base/test_image.py
+++ b/tests/unit/data/base/test_image.py
@@ -4,7 +4,9 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import pytest
+
 from anomalib.data import AnomalibDataModule
+
 from .test_datamodule import _TestAnomalibDataModule
 
 
diff --git a/tests/unit/data/base/test_video.py b/tests/unit/data/base/test_video.py
index f862223614..c93e9d1eaf 100644
--- a/tests/unit/data/base/test_video.py
+++ b/tests/unit/data/base/test_video.py
@@ -4,7 +4,9 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import pytest
+
 from anomalib.data import AnomalibDataModule
+
 from .test_datamodule import _TestAnomalibDataModule
 
 
diff --git a/tests/unit/data/conftest.py b/tests/unit/data/conftest.py
index c60b985e1d..8726b07849 100644
--- a/tests/unit/data/conftest.py
+++ b/tests/unit/data/conftest.py
@@ -3,9 +3,10 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import TaskType
 import pytest
 
+from anomalib.data import TaskType
+
 
 @pytest.fixture(params=[TaskType.CLASSIFICATION, TaskType.DETECTION, TaskType.SEGMENTATION])
 def task_type(request) -> str:
diff --git a/tests/unit/data/test_avenue.py b/tests/unit/data/test_avenue.py
index 4bd2fb2049..d09db3bea1 100644
--- a/tests/unit/data/test_avenue.py
+++ b/tests/unit/data/test_avenue.py
@@ -4,12 +4,14 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import os
-from anomalib.data import Avenue, TaskType
+
 import pytest
-from .base import _TestAnomalibVideoDatamodule
 
+from anomalib.data import Avenue, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibVideoDatamodule
+
 
 class TestAvenue(_TestAnomalibVideoDatamodule):
     """Avenue Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_btech.py b/tests/unit/data/test_btech.py
index 5597344128..aede60ad34 100644
--- a/tests/unit/data/test_btech.py
+++ b/tests/unit/data/test_btech.py
@@ -3,12 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import BTech, TaskType
 import pytest
-from .base import _TestAnomalibImageDatamodule
 
+from anomalib.data import BTech, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibImageDatamodule
+
 
 class TestBTech(_TestAnomalibImageDatamodule):
     """MVTec Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_folder.py b/tests/unit/data/test_folder.py
index e50e487bd1..83a0e0343a 100644
--- a/tests/unit/data/test_folder.py
+++ b/tests/unit/data/test_folder.py
@@ -3,12 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import Folder, TaskType
-from .base import _TestAnomalibImageDatamodule
 import pytest
 
+from anomalib.data import Folder, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibImageDatamodule
+
 
 class TestFolder(_TestAnomalibImageDatamodule):
     """Folder Datamodule Unit Tests.
diff --git a/tests/unit/data/test_folder_3d.py b/tests/unit/data/test_folder_3d.py
index 73c19a8343..04e2fa9d73 100644
--- a/tests/unit/data/test_folder_3d.py
+++ b/tests/unit/data/test_folder_3d.py
@@ -3,12 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import Folder3D, TaskType
 import pytest
-from .base import _TestAnomalibDepthDatamodule
 
+from anomalib.data import Folder3D, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibDepthDatamodule
+
 
 class TestFolder3D(_TestAnomalibDepthDatamodule):
     """Folder3D Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_mvtec.py b/tests/unit/data/test_mvtec.py
index 7a3a91207b..7bea523f06 100644
--- a/tests/unit/data/test_mvtec.py
+++ b/tests/unit/data/test_mvtec.py
@@ -3,12 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import MVTec, TaskType
 import pytest
-from .base import _TestAnomalibImageDatamodule
 
+from anomalib.data import MVTec, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibImageDatamodule
+
 
 class TestMVTec(_TestAnomalibImageDatamodule):
     """MVTec Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_mvtec_3d.py b/tests/unit/data/test_mvtec_3d.py
index 33f9d1a178..42060be56b 100644
--- a/tests/unit/data/test_mvtec_3d.py
+++ b/tests/unit/data/test_mvtec_3d.py
@@ -3,12 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import MVTec3D, TaskType
 import pytest
-from .base import _TestAnomalibDepthDatamodule
 
+from anomalib.data import MVTec3D, TaskType
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibDepthDatamodule
+
 
 class TestMVTec3D(_TestAnomalibDepthDatamodule):
     """MVTec Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_shanghaitech.py b/tests/unit/data/test_shanghaitech.py
index c7cee9c80f..968988aaef 100644
--- a/tests/unit/data/test_shanghaitech.py
+++ b/tests/unit/data/test_shanghaitech.py
@@ -3,14 +3,14 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import ShanghaiTech, TaskType
 import pytest
 
+from anomalib.data import ShanghaiTech, TaskType
 from anomalib.data.utils.split import ValSplitMode
-from .base import _TestAnomalibVideoDatamodule
-
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibVideoDatamodule
+
 
 class TestShanghaiTech(_TestAnomalibVideoDatamodule):
     """ShanghaiTech Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_ucsd_ped.py b/tests/unit/data/test_ucsd_ped.py
index ef569bb34b..45a68d4bbd 100644
--- a/tests/unit/data/test_ucsd_ped.py
+++ b/tests/unit/data/test_ucsd_ped.py
@@ -3,13 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import UCSDped, TaskType
 import pytest
-from .base import _TestAnomalibVideoDatamodule
-
 
+from anomalib.data import TaskType, UCSDped
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibVideoDatamodule
+
 
 class TestUCSDped(_TestAnomalibVideoDatamodule):
     """UCSDped Datamodule Unit Tests."""
diff --git a/tests/unit/data/test_visa.py b/tests/unit/data/test_visa.py
index 8e983c7c09..4b64843593 100644
--- a/tests/unit/data/test_visa.py
+++ b/tests/unit/data/test_visa.py
@@ -3,13 +3,13 @@
 # Copyright (C) 2023 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from anomalib.data import Visa, TaskType
 import pytest
-from .base import _TestAnomalibImageDatamodule
-
 
+from anomalib.data import TaskType, Visa
 from tests.helpers.dataset import get_dataset_path
 
+from .base import _TestAnomalibImageDatamodule
+
 
 class TestVisa(_TestAnomalibImageDatamodule):
     """Visa Datamodule Unit Tests."""