From 423210d1b0ad083aba8535ae2bd8dbe2d62f7cfa Mon Sep 17 00:00:00 2001
From: Paulo Valente <16843419+polvalente@users.noreply.github.com>
Date: Thu, 22 Aug 2024 13:33:27 -0300
Subject: [PATCH] fix: small livebook fixes

---
 guides/gridworld.livemd | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/guides/gridworld.livemd b/guides/gridworld.livemd
index 82b1198..29ebc82 100644
--- a/guides/gridworld.livemd
+++ b/guides/gridworld.livemd
@@ -79,7 +79,7 @@ max_points = 1000
 
 plot_fn = fn axon_state ->
   if axon_state.iteration > 1 do
-    episode = axon_state.epoch
+    episode = axon_state.episode
 
     Kino.VegaLite.clear(grid_widget, dataset: "target")
     Kino.VegaLite.clear(grid_widget, dataset: "trajectory")
@@ -158,6 +158,7 @@ state_space_shape = {delta_x, delta_y, delta_x, delta_y}
        exploration_eps: 1.0e-4},
       plot_fn,
       state_to_trajectory_fn,
+      checkpoint_path: "/tmp/gridworld",
       num_episodes: episodes,
       max_iter: max_iter
     )