Update example notebook

zombie-einstein · May 13, 2023 · 5d43dd7 · 5d43dd7
1 parent b34e65c
commit 5d43dd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",