run ruff --fix on unused imports

jscanvic · jscanvic · commit 8ff5657a299c · 2025-06-20T14:38:51.000+02:00
diff --git a/deepinv/datasets/fastmri.py b/deepinv/datasets/fastmri.py
@@ -22,12 +22,10 @@
 from typing import Any, Callable, NamedTuple, Optional, Union, Any
 from collections import defaultdict
 import pickle
-import math
 import warnings
 import os
 import h5py
 from tqdm import tqdm
-import numpy as np
 import torch
 from torchvision.transforms import Compose, CenterCrop
 
diff --git a/deepinv/datasets/lidc_idri.py b/deepinv/datasets/lidc_idri.py
@@ -6,7 +6,6 @@
 )
 import os
 
-import numpy
 import torch
 import numpy as np
 
diff --git a/deepinv/loss/adversarial/uair.py b/deepinv/loss/adversarial/uair.py
@@ -1,6 +1,6 @@
 import torch.nn as nn
 from torch import Tensor
-from .base import GeneratorLoss, DiscriminatorLoss
+from .base import GeneratorLoss
 from deepinv.physics import Physics
 
 
diff --git a/deepinv/loss/augmentation.py b/deepinv/loss/augmentation.py
@@ -13,7 +13,6 @@
 from deepinv.transform.base import Transform, Identity
 from deepinv.transform.rotate import Rotate
 from deepinv.transform.shift import Shift
-from deepinv.physics.forward import LinearPhysics
 
 
 class AugmentConsistencyLoss(Loss):
diff --git a/deepinv/loss/mc.py b/deepinv/loss/mc.py
@@ -3,7 +3,6 @@
 import torch
 from deepinv.loss.loss import Loss
 from deepinv.loss.metric.metric import Metric
-from deepinv.transform.base import Transform
 
 
 class MCLoss(Loss):
diff --git a/deepinv/loss/metric/distortion.py b/deepinv/loss/metric/distortion.py
@@ -4,7 +4,6 @@
 
 import torch
 from torch import Tensor
-from torch.nn import MSELoss, L1Loss
 from torchmetrics.functional.image import (
     structural_similarity_index_measure,
     multiscale_structural_similarity_index_measure,
diff --git a/deepinv/loss/moi.py b/deepinv/loss/moi.py
@@ -1,7 +1,6 @@
 from typing import Union, Optional
 from copy import deepcopy
 
-import numpy as np
 import torch
 
 from deepinv.loss.loss import Loss
diff --git a/deepinv/loss/score.py b/deepinv/loss/score.py
@@ -1,5 +1,4 @@
 import torch
-import deepinv.physics
 from deepinv.loss.loss import Loss
 from deepinv.models.base import Reconstructor
 
diff --git a/deepinv/loss/sure.py b/deepinv/loss/sure.py
@@ -1,7 +1,5 @@
 import torch
-import torch.nn as nn
 import numpy as np
-import deepinv.physics
 from deepinv.loss.loss import Loss
 
 
diff --git a/deepinv/models/epll.py b/deepinv/models/epll.py
@@ -1,4 +1,3 @@
-import torch
 from deepinv.optim.epll import EPLL
 from deepinv.physics import Denoising, GaussianNoise
 from .base import Denoiser
diff --git a/deepinv/models/equivariant.py b/deepinv/models/equivariant.py
@@ -1,5 +1,4 @@
 from typing import Optional
-import torch
 from deepinv.transform import Transform, Rotate, Reflect
 from .base import Denoiser
 
diff --git a/deepinv/models/icnn.py b/deepinv/models/icnn.py
@@ -1,5 +1,4 @@
 # Code borrowed from https://github.com/ZhenghanFang/learned-proximal-networks
-import numpy as np
 import torch
 from torch import nn
 
diff --git a/deepinv/models/varnet.py b/deepinv/models/varnet.py
@@ -12,7 +12,7 @@
 from deepinv.physics.mri import MRIMixin, MRI, MultiCoilMRI
 
 if TYPE_CHECKING:
-    from deepinv.physics.forward import Physics
+    pass
 
 
 class VarNet(ArtifactRemoval, MRIMixin):
diff --git a/deepinv/optim/bregman.py b/deepinv/optim/bregman.py
@@ -1,5 +1,4 @@
 import torch
-import torch.nn as nn
 from deepinv.optim.potential import Potential
 
 
diff --git a/deepinv/optim/distance.py b/deepinv/optim/distance.py
@@ -1,6 +1,5 @@
 import torch
 from deepinv.optim.potential import Potential
-from deepinv.utils.tensorlist import TensorList
 
 
 class Distance(Potential):
diff --git a/deepinv/optim/optim_iterators/optim_iterator.py b/deepinv/optim/optim_iterators/optim_iterator.py
@@ -1,4 +1,3 @@
-import torch
 import torch.nn as nn
 
 
diff --git a/deepinv/optim/optim_iterators/primal_dual_CP.py b/deepinv/optim/optim_iterators/primal_dual_CP.py
@@ -1,4 +1,3 @@
-import torch
 
 from .optim_iterator import OptimIterator, fStep, gStep
 
diff --git a/deepinv/optim/prior.py b/deepinv/optim/prior.py
@@ -13,7 +13,6 @@
 from deepinv.models.tv import TVDenoiser
 from deepinv.models.wavdict import WaveletDenoiser, WaveletDictDenoiser
 from deepinv.utils import patch_extractor
-from typing import Callable
 
 
 class Prior(Potential):
diff --git a/deepinv/physics/blur.py b/deepinv/physics/blur.py
@@ -1,7 +1,6 @@
 from torchvision.transforms.functional import rotate
 import torchvision
 import torch
-import numpy as np
 import torch.fft as fft
 from torch import Tensor
 from deepinv.physics.forward import LinearPhysics, DecomposablePhysics
diff --git a/deepinv/physics/forward.py b/deepinv/physics/forward.py
@@ -1,5 +1,5 @@
 from __future__ import annotations
-from typing import Union, List
+from typing import Union
 
 import torch
 from torch import Tensor
diff --git a/deepinv/physics/functional/astra.py b/deepinv/physics/functional/astra.py
@@ -5,7 +5,6 @@
 
 try:
     import astra
-    from astra import experimental
 except:
     astra = ImportError("The astra-toolbox package is not installed.")
 
diff --git a/deepinv/physics/generator/base.py b/deepinv/physics/generator/base.py
@@ -1,7 +1,6 @@
 import torch
 import torch.nn as nn
 from typing import Union
-import warnings
 from hashlib import sha256
 
 
diff --git a/deepinv/sampling/diffusion.py b/deepinv/sampling/diffusion.py
@@ -5,7 +5,6 @@
 
 import deepinv.physics
 from deepinv.sampling.langevin import MonteCarlo
-from deepinv.utils.plotting import plot
 
 
 class DiffusionSampler(MonteCarlo):
diff --git a/deepinv/sampling/langevin.py b/deepinv/sampling/langevin.py
@@ -6,7 +6,7 @@
 import deepinv.optim
 from tqdm import tqdm
 from deepinv.optim.utils import check_conv
-from deepinv.sampling.utils import Welford, projbox, refl_projbox
+from deepinv.sampling.utils import Welford, projbox
 
 
 class MonteCarlo(nn.Module):
diff --git a/deepinv/tests/test_adversarial.py b/deepinv/tests/test_adversarial.py
@@ -5,7 +5,6 @@
 
 import deepinv as dinv
 from deepinv.loss import adversarial
-from test_loss import dataset, physics
 
 ADVERSARIAL_COMBOS = ["DeblurGAN", "CSGM", "AmbientGAN", "UAIR"]
 
diff --git a/deepinv/tests/test_datasets.py b/deepinv/tests/test_datasets.py
@@ -30,7 +30,6 @@
 from deepinv.physics.generator import GaussianMaskGenerator
 
 from unittest.mock import patch
-import unittest.mock as mock
 import io
 
 
@@ -591,7 +590,6 @@ def download_CMRxRecon():
 
 
 def test_CMRxReconSliceDataset(download_CMRxRecon):
-    from math import prod
 
     img_size = (12, 512, 256)
 
diff --git a/deepinv/tests/test_loss_train.py b/deepinv/tests/test_loss_train.py
@@ -8,7 +8,7 @@
 import deepinv as dinv
 from deepinv.optim.data_fidelity import L2
 from deepinv.optim.prior import PnP
-from dummy import DummyCircles, DummyModel
+from dummy import DummyCircles
 from deepinv.unfolded import unfolded_builder
 from deepinv.physics import Inpainting, GaussianNoise, Blur, Pansharpen
 from deepinv.physics.generator import (
diff --git a/deepinv/tests/test_physics.py b/deepinv/tests/test_physics.py
@@ -9,7 +9,7 @@
 from deepinv.physics.forward import adjoint_function
 import deepinv as dinv
 from deepinv.optim.data_fidelity import L2
-from deepinv.physics.mri import MRI, MRIMixin, DynamicMRI, MultiCoilMRI, SequentialMRI
+from deepinv.physics.mri import MRI, MRIMixin, DynamicMRI, MultiCoilMRI
 from deepinv.utils import TensorList
 
 
diff --git a/deepinv/tests/test_physics_functional.py b/deepinv/tests/test_physics_functional.py
@@ -8,7 +8,6 @@
 
 import deepinv as dinv
 import torch
-import pytest
 
 
 def test_conv2d_adjointness(device):
diff --git a/deepinv/tests/test_trainer.py b/deepinv/tests/test_trainer.py
@@ -12,7 +12,6 @@
 
 from unittest.mock import patch
 import math
-import builtins
 import io
 import contextlib
 import re
diff --git a/deepinv/training/testing.py b/deepinv/training/testing.py
@@ -1,4 +1,3 @@
-import torch
 from deepinv.loss.metric import PSNR
 from deepinv.training import Trainer
 
diff --git a/deepinv/transform/reflect.py b/deepinv/transform/reflect.py
@@ -1,8 +1,5 @@
 from typing import Union, Iterable
 import torch
-from torchvision.transforms.functional import rotate
-from torchvision.transforms import InterpolationMode
-import numpy as np
 from deepinv.transform.base import Transform, TransformParam
 import itertools
 
diff --git a/deepinv/utils/demo.py b/deepinv/utils/demo.py
@@ -9,11 +9,9 @@
 import numpy as np
 import torch
 from torch.utils.data import Dataset
-from torch.nn import Module
 import torchvision
 from torchvision import transforms
 
-from deepinv.models.base import Reconstructor, Denoiser
 
 
 def get_git_root():
diff --git a/deepinv/utils/metric.py b/deepinv/utils/metric.py
@@ -1,5 +1,3 @@
-import torch
-import numpy as np
 import warnings
 
 
diff --git a/deepinv/utils/parameters.py b/deepinv/utils/parameters.py
@@ -1,4 +1,3 @@
-import numpy as np
 
 
 def get_GSPnP_params(problem, noise_level_img):
diff --git a/deepinv/utils/plotting.py b/deepinv/utils/plotting.py
@@ -21,7 +21,6 @@
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 
 from PIL import Image
-import io
 
 
 def config_matplotlib(fontsize=17):
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -9,7 +9,6 @@
 from sphinx_gallery.directives import ImageSg
 import sys
 import os
-from pathlib import Path
 from sphinx.util import logging
 
 logger = logging.getLogger(__name__)
diff --git a/examples/adversarial-learning/demo_gan_imaging.py b/examples/adversarial-learning/demo_gan_imaging.py
@@ -39,9 +39,7 @@
 
 import torch
 from torch.utils.data import DataLoader, random_split
-from torchvision.datasets import ImageFolder
 from torchvision.transforms import Compose, ToTensor, CenterCrop, Resize
-from torchvision.datasets.utils import download_and_extract_archive
 
 import deepinv as dinv
 from deepinv.loss import adversarial
diff --git a/examples/basics/demo_phase_retrieval.py b/examples/basics/demo_phase_retrieval.py
@@ -27,7 +27,6 @@
 from deepinv.optim.phase_retrieval import (
     correct_global_phase,
     cosine_similarity,
-    spectral_methods,
 )
 from deepinv.models.complex import to_complex_denoiser
 
diff --git a/examples/basics/demo_spc.py b/examples/basics/demo_spc.py
@@ -31,7 +31,6 @@
 
 import torch
 from pathlib import Path
-import matplotlib.pyplot as plt
 import deepinv as dinv
 from deepinv.utils.demo import get_image_url, load_url_image
 from deepinv.utils.plotting import plot
diff --git a/examples/optimization/demo_3D_wavelets.py b/examples/optimization/demo_3D_wavelets.py
@@ -10,7 +10,6 @@
 import deepinv as dinv
 from pathlib import Path
 import numpy as np
-import matplotlib.pyplot as plt
 
 import torch
 import torch.nn as nn
diff --git a/examples/optimization/demo_wavelet_prior.py b/examples/optimization/demo_wavelet_prior.py
@@ -15,7 +15,7 @@
 
 from deepinv.optim.data_fidelity import L2
 from deepinv.optim.optimizers import optim_builder
-from deepinv.utils.demo import load_dataset, load_degradation
+from deepinv.utils.demo import load_dataset
 from deepinv.utils.plotting import plot, plot_curves
 
 # %%
diff --git a/examples/plug-and-play/demo_PnP_mirror_descent.py b/examples/plug-and-play/demo_PnP_mirror_descent.py
@@ -19,7 +19,6 @@
 import deepinv as dinv
 from pathlib import Path
 import torch
-from torch.utils.data import DataLoader
 from deepinv.optim.data_fidelity import PoissonLikelihood
 from deepinv.optim.prior import RED
 from deepinv.optim import optim_builder
diff --git a/examples/self-supervised-learning/demo_artifact2artifact.py b/examples/self-supervised-learning/demo_artifact2artifact.py
@@ -24,9 +24,8 @@
 
 """
 
-from pathlib import Path
 import torch
-from torch.utils.data import DataLoader, Subset
+from torch.utils.data import DataLoader
 from torchvision import transforms
 
 import deepinv as dinv
diff --git a/examples/self-supervised-learning/demo_r2r_denoising.py b/examples/self-supervised-learning/demo_r2r_denoising.py
@@ -16,7 +16,6 @@
 
 import deepinv as dinv
 from deepinv.utils.demo import get_data_home
-from deepinv.models.utils import get_weights_url
 
 # %%
 # Setup paths for data loading and results.
diff --git a/examples/unfolded/demo_DEQ.py b/examples/unfolded/demo_DEQ.py
@@ -14,11 +14,9 @@
 from pathlib import Path
 import torch
 from torch.utils.data import DataLoader
-from deepinv.models import DnCNN
 from deepinv.optim.data_fidelity import L2
 from deepinv.optim.prior import PnP
 from deepinv.unfolded import DEQ_builder
-from deepinv.training import train, test
 from torchvision import transforms
 from deepinv.utils.demo import load_dataset, load_degradation
 

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,6 @@`
`6`	`6`	`)`
`7`	`7`	`import os`
`8`	`8`
`9`		`-import numpy`
`10`	`9`	`import torch`
`11`	`10`	`import numpy as np`
`12`	`11`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`import torch`
`2`		`-import deepinv.physics`
`3`	`2`	`from deepinv.loss.loss import Loss`
`4`	`3`	`from deepinv.models.base import Reconstructor`
`5`	`4`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-import torch`
`2`	`1`	`from deepinv.optim.epll import EPLL`
`3`	`2`	`from deepinv.physics import Denoising, GaussianNoise`
`4`	`3`	`from .base import Denoiser`