small change in docstring

lightly-ai · Jan 10, 2025 · e692963 · e692963
1 parent 378abce
commit e692963
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lightly/models/utils.py b/lightly/models/utils.py
@@ -40,9 +40,9 @@ 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`
-            corresponds to `num_cls`.
+    Returns: 
+        A tensor of shape :math:`(B, C, N)` or :math:`(C, N)` where :math:`N` 
+        corresponds to `num_cls`.
     """
     if source.dim() == 3:
         return _mask_reduce(source, mask, reduce, num_cls)