From 8ecdbcadb877e77e1bca73e32a6ccac7ba14807e Mon Sep 17 00:00:00 2001
From: russellizadi <russell.izadi@gmail.com>
Date: Tue, 19 Aug 2025 18:08:06 -0400
Subject: [PATCH] Remove state return from function example in compile
 documentation

---
 docs/src/usage/compile.rst | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/src/usage/compile.rst b/docs/src/usage/compile.rst
index 7fe0ffd4f..ae01bb1f3 100644
--- a/docs/src/usage/compile.rst
+++ b/docs/src/usage/compile.rst
@@ -225,7 +225,7 @@ In some cases returning updated state can be pretty inconvenient. Hence,
   def fun(x, y):
       z = x + y
       state.append(z)
-      return mx.exp(z), state
+      return mx.exp(z)
 
   fun(mx.array(1.0), mx.array(2.0))
   # Prints [array(3, dtype=float32)]