fix: small livebook fixes

DockYard · Aug 22, 2024 · 423210d · 423210d
1 parent 9dcd1b4
commit 423210d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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
     )