Problem in reproducing demo: Poisson equation in 1D with Dirichlet boundary conditions

[mickpires]

I ain't being able to reproduce the following step:
model.restore(f"model/model-{train_state.best_step}.ckpt", verbose=1)

It says ValueError: The passed save_path is not a valid checkpoint: model/model-9000.ckpt.

I have done a ls on the folder model and Imma copy here

Mode LastWriteTime Length Name

---

-a---- 20/06/2025 20:17 97 checkpoint
-a---- 20/06/2025 20:17 21004 model.ckpt-1.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-1.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-1.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-1000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-1000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-1000.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-2000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-2000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-2000.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-3000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-3000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-3000.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-6000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-6000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-6000.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-7000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-7000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-7000.ckpt.meta
-a---- 20/06/2025 20:17 21004 model.ckpt-9000.ckpt.data-00000-of-00001
-a---- 20/06/2025 20:17 348 model.ckpt-9000.ckpt.index
-a---- 20/06/2025 20:17 181856 model.ckpt-9000.ckpt.meta