move SMoGPrototypes docstring to correct indent

lightly-ai · Oct 18, 2024 · 166a46a · 166a46a
1 parent d4fc1bf
commit 166a46a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/lightly/models/modules/heads.py b/lightly/models/modules/heads.py
@@ -350,14 +350,14 @@ def __init__(
 
 
 class SMoGPrototypes(nn.Module):
-        """SMoG prototypes module for synchronous momentum grouping.
+    """SMoG prototypes module for synchronous momentum grouping.
 
-        Args:
-            group_features: 
-                Tensor containing the group features.
-            beta: 
-                Beta parameter for momentum updating.
-        """
+    Args:
+        group_features: 
+            Tensor containing the group features.
+        beta: 
+            Beta parameter for momentum updating.
+    """
 
     def __init__(
         self,