fix formatting

lightly-ai · Nov 7, 2024 · f538d1d · f538d1d
1 parent 9f09fa8
commit f538d1d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lightly/transforms/rotation.py b/lightly/transforms/rotation.py
@@ -66,9 +66,9 @@ class RandomRotateDegrees:
     """
 
     def __init__(self, prob: float, degrees: Union[float, Tuple[float, float]]):
-        self.transform: Callable[[Union[Image, Tensor]], Union[Image, Tensor]] = (
-            T.RandomApply([T.RandomRotation(degrees=degrees)], p=prob)
-        )
+        self.transform: Callable[
+            [Union[Image, Tensor]], Union[Image, Tensor]
+        ] = T.RandomApply([T.RandomRotation(degrees=degrees)], p=prob)
 
     def __call__(self, image: Union[Image, Tensor]) -> Union[Image, Tensor]:
         """Rotates the images with a given probability.