From 5d43dd70a7aaf7220e2730a1a8043b1558b3c466 Mon Sep 17 00:00:00 2001
From: zombie-einstein <zombie-einstein@proton.me>
Date: Sat, 13 May 2023 15:46:48 +0100
Subject: [PATCH] Update example notebook

---
 examples/lstm_usage.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/lstm_usage.ipynb b/examples/lstm_usage.ipynb
index 403ee04..bf0edae 100644
--- a/examples/lstm_usage.ipynb
+++ b/examples/lstm_usage.ipynb
@@ -149,7 +149,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "_, agent, hidden_states = jax_ppo.init_lstm_agent(\n",
+    "_, agent = jax_ppo.init_lstm_agent(\n",
     "    k, \n",
     "    params,\n",
     "    env.action_space().shape,\n",