another formatting error

lightly-ai · Jan 10, 2025 · d35cbd1 · d35cbd1
1 parent e692963
commit d35cbd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightly/models/utils.py b/lightly/models/utils.py
@@ -40,8 +40,8 @@ def pool_masked(
             integer indices.
         num_cls: The number of classes in the possible masks.
 
-    Returns: 
-        A tensor of shape :math:`(B, C, N)` or :math:`(C, N)` where :math:`N` 
+    Returns:
+        A tensor of shape :math:`(B, C, N)` or :math:`(C, N)` where :math:`N`
         corresponds to `num_cls`.
     """
     if source.dim() == 3: