From 07e06bff441f4e737e0ea131de1e1e043edc5d8a Mon Sep 17 00:00:00 2001
From: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com>
Date: Fri, 4 Oct 2024 22:48:16 +0000
Subject: [PATCH] Add nnx.graph.flatten to graph.rst autosummary

---
 docs_nnx/api_reference/flax.nnx/graph.rst | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs_nnx/api_reference/flax.nnx/graph.rst b/docs_nnx/api_reference/flax.nnx/graph.rst
index d944e3c7bf..763064b736 100644
--- a/docs_nnx/api_reference/flax.nnx/graph.rst
+++ b/docs_nnx/api_reference/flax.nnx/graph.rst
@@ -14,6 +14,7 @@ graph
 .. autofunction:: iter_graph
 .. autofunction:: clone
 .. autofunction:: call
+.. autofunction:: flatten
 
 .. autoclass:: GraphDef
   :members:
@@ -22,4 +23,4 @@ graph
   :members:
 
 .. autofunction:: update_context
-.. autofunction:: current_update_context
\ No newline at end of file
+.. autofunction:: current_update_context