docs: docstrings for Identity module

ml-gde · Sep 13, 2024 · 52776d6 · 52776d6
1 parent 09d1e08
commit 52776d6
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/jflux/layers.py b/jflux/layers.py
@@ -206,15 +206,7 @@ def __call__(self, z: Array) -> Array:
 
 
 class Identity(nnx.Module):
-    """
-    Identity module.
-
-    Args:
-        x (Array): Input array.
-
-    Returns:
-        Array: The input array.
-    """
+    """Identity module."""
 
     def __call__(self, x: Array) -> Array:
         return x