From 52776d6931498dee0f983098a772d04ba4583859 Mon Sep 17 00:00:00 2001
From: Saurav Maheshkar <sauravvmaheshkar@gmail.com>
Date: Fri, 13 Sep 2024 04:59:27 +0100
Subject: [PATCH] docs: docstrings for Identity module

---
 jflux/layers.py | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/jflux/layers.py b/jflux/layers.py
index cfd2be8..d0a6d15 100644
--- 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