refactor: update naming and remove unused package from AmplitudeRescaleTransform

lightly/transforms/__init__.py

@@ -9,7 +9,7 @@
 # All Rights Reserved
 
 from lightly.transforms.aim_transform import AIMTransform
-from lightly.transforms.amplitude_rescale_transform import AmplitudeRescaleTranform
+from lightly.transforms.amplitude_rescale_transform import AmplitudeRescaleTransform
 from lightly.transforms.byol_transform import (
     BYOLTransform,
     BYOLView1Transform,

lightly/transforms/amplitude_rescale_transform.py

@@ -1,12 +1,11 @@
 from typing import Tuple
 
-import numpy as np
 import torch
 from torch import Tensor
 from torch.distributions import Uniform
 
 
-class AmplitudeRescaleTranform:
+class AmplitudeRescaleTransform:
     """Implementation of amplitude rescaling transformation.
 
     This transform will rescale the amplitude of the Fourier Spectrum (`freq_image`) of the image and return it.

tests/transforms/test_amplitude_rescale_transform.py

@@ -1,9 +1,7 @@
-import numpy as np
 import torch
 
 from lightly.transforms import (
-    AmplitudeRescaleTranform,
-    IRFFT2DTransform,
+    AmplitudeRescaleTransform,
     RFFT2DTransform,
 )
 
@@ -16,10 +14,10 @@ def test() -> None:
     rfftTransform = RFFT2DTransform()
     rfft = rfftTransform(image)
 
-    ampRescaleTf_1 = AmplitudeRescaleTranform()
+    ampRescaleTf_1 = AmplitudeRescaleTransform()
     rescaled_rfft_1 = ampRescaleTf_1(rfft)
 
-    ampRescaleTf_2 = AmplitudeRescaleTranform(range=(1.0, 2.0))
+    ampRescaleTf_2 = AmplitudeRescaleTransform(range=(1.0, 2.0))
     rescaled_rfft_2 = ampRescaleTf_2(rfft)
 
     assert rescaled_rfft_1.shape == rfft.shape