From 96afe4f066912ebb50db8227c99a87d6c8041e15 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 21 Aug 2024 17:58:02 +0200
Subject: [PATCH] [qa] Improve ruff/isort

---
 helpers/generate_histograms.py                      | 3 ++-
 helpers/montgomery_preprocess.py                    | 5 +++--
 tests/classify/test_saliencymap_interpretability.py | 3 ++-
 tests/conftest.py                                   | 1 +
 tests/segment/test_measures.py                      | 1 +
 tests/test_image_utils.py                           | 1 +
 tests/test_transforms.py                            | 1 +
 7 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/helpers/generate_histograms.py b/helpers/generate_histograms.py
index 84cff565..aefc170d 100755
--- a/helpers/generate_histograms.py
+++ b/helpers/generate_histograms.py
@@ -26,10 +26,11 @@ import PIL.Image
 import torch
 from clapper.click import ResourceOption
 from clapper.logging import setup
+from torchvision.transforms.functional import to_pil_image
+
 from mednet.data.typing import Dataset
 from mednet.scripts.click import ConfigCommand
 from mednet.scripts.database import _get_raw_databases
-from torchvision.transforms.functional import to_pil_image
 
 logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s")
 
diff --git a/helpers/montgomery_preprocess.py b/helpers/montgomery_preprocess.py
index 399c524b..6699a75f 100644
--- a/helpers/montgomery_preprocess.py
+++ b/helpers/montgomery_preprocess.py
@@ -12,11 +12,12 @@ import sys
 
 import numpy
 import PIL.Image
-from mednet.data.image_utils import remove_black_borders
-from mednet.models.transforms import square_center_pad
 from torch.nn.functional import interpolate
 from torchvision.transforms.functional import resize, to_pil_image, to_tensor
 
+from mednet.data.image_utils import remove_black_borders
+from mednet.models.transforms import square_center_pad
+
 
 def main():
     if len(sys.argv) != 2:
diff --git a/tests/classify/test_saliencymap_interpretability.py b/tests/classify/test_saliencymap_interpretability.py
index 90889403..e741230b 100644
--- a/tests/classify/test_saliencymap_interpretability.py
+++ b/tests/classify/test_saliencymap_interpretability.py
@@ -2,13 +2,14 @@
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
 import numpy as np
+from torchvision import tv_tensors
+
 from mednet.engine.classify.saliency.interpretability import (
     _compute_avg_saliency_focus,
     _compute_binary_mask,
     _compute_proportional_energy,
     _process_sample,
 )
-from torchvision import tv_tensors
 
 
 def test_compute_avg_saliency_focus():
diff --git a/tests/conftest.py b/tests/conftest.py
index 8cd41d58..9260716b 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -8,6 +8,7 @@ import typing
 import numpy
 import pytest
 import torch
+
 from mednet.data.split import JSONDatabaseSplit
 from mednet.data.typing import DatabaseSplit
 
diff --git a/tests/segment/test_measures.py b/tests/segment/test_measures.py
index 66578a76..2f12d927 100644
--- a/tests/segment/test_measures.py
+++ b/tests/segment/test_measures.py
@@ -5,6 +5,7 @@
 import random
 
 import numpy
+
 from mednet.engine.segment.evaluator import all_metrics
 
 
diff --git a/tests/test_image_utils.py b/tests/test_image_utils.py
index ab431eca..47da8781 100644
--- a/tests/test_image_utils.py
+++ b/tests/test_image_utils.py
@@ -5,6 +5,7 @@
 
 import numpy
 import PIL.Image
+
 from mednet.data.image_utils import remove_black_borders
 
 
diff --git a/tests/test_transforms.py b/tests/test_transforms.py
index 8bbfc850..700835d8 100644
--- a/tests/test_transforms.py
+++ b/tests/test_transforms.py
@@ -7,6 +7,7 @@ import numpy
 import PIL.Image
 import torch
 import torchvision.transforms.functional as F  # noqa: N812
+
 from mednet.data.augmentations import ElasticDeformation
 from mednet.models.transforms import crop_image_to_mask
 
-- 
GitLab