refactor: reformatting

lightly-ai · Nov 29, 2024 · da8dd51 · da8dd51
1 parent 658dd83
commit da8dd51
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lightly/models/utils.py b/lightly/models/utils.py
@@ -1276,9 +1276,7 @@ def repeat_interleave_batch(x: Tensor, B: int, repeat: int) -> Tensor:
     return x
 
 
-def apply_masks(
-    x: Tensor, masks: Tensor | list[Tensor]
-) -> Tensor:
+def apply_masks(x: Tensor, masks: Tensor | list[Tensor]) -> Tensor:
     """Apply masks to the input tensor.
 
     From https://github.com/facebookresearch/ijepa/blob/main/src/masks/utils.py