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