From a6a267ecb7c8f31acfbf1a927bfcbaeaf792881c Mon Sep 17 00:00:00 2001
From: ASEM000 <mahmoudasem00@gmail.com>
Date: Fri, 18 Aug 2023 01:21:07 +0300
Subject: [PATCH] Update mental_model.ipynb

---
 docs/notebooks/mental_model.ipynb | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/docs/notebooks/mental_model.ipynb b/docs/notebooks/mental_model.ipynb
index 5ccaf00..18a6d9b 100644
--- a/docs/notebooks/mental_model.ipynb
+++ b/docs/notebooks/mental_model.ipynb
@@ -11,7 +11,6 @@
     "\n",
     "3) **Transformation Transparent Code**: Write code that works under any `jax` transformation. This approach makes the code reusable, easier to understand, and less error-prone.\n",
     "\n",
-    "\n",
     "The following code showcases these principles"
    ]
   },
@@ -40,8 +39,6 @@
     "# to showcase the functional update of with stateful modules\n",
     "# lets define a tree of modules with stateful updates in the call method\n",
     "\n",
-    "\n",
-    "@sk.autoinit\n",
     "class Linears(sk.TreeClass):\n",
     "    def __init__(self):\n",
     "        self.l1 = Linear(1, 120, key=jax.random.PRNGKey(0))\n",