diff --git a/helpers/generate_histograms.py b/helpers/generate_histograms.py index 84cff565228bdd307ef0f8bdb9359e7871e08b85..aefc170dcf09138d293889bc70355a3fec03519c 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 399c524b40bbf1846e0177fa36b5fc339d675618..6699a75f7582b44598bdf135cc4379fda6ff0e13 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 90889403f9ae7ea96beb2b4835b04b6372ef6ad2..e741230bcd8bb400b1472a6c95714ce3475caf7f 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 8cd41d58029250618808035c2b91a5b38238a8bc..9260716b9dd2a524734b721c147c879cdb66d483 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 66578a76bd2afb08446d700f6b6bbe28e8dd305e..2f12d9279328e0692e8ce376f7a95c5fb11f00d9 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 ab431eca66f4170278363e780bdb960799b425ce..47da87818db3161a39d3579cedf51e0bc2d7de2a 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 8bbfc85032380dcab45a1df433c4a301009b2088..700835d82f14256302b26aa772677fa0c64a97fb 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