From f0033789638bc72d0215a312d846d4425b3aec74 Mon Sep 17 00:00:00 2001
From: Noah Luna <15202580+ngrayluna@users.noreply.github.com>
Date: Tue, 2 Dec 2025 13:37:29 -0800
Subject: [PATCH] Keras: Update Wandb imports and model checkpoint file
 extension

---
 colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb b/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
index 6c3ada81..7280dbec 100644
--- a/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
+++ b/colabs/intro/Intro_to_Weights_&_Biases_keras.ipynb
@@ -160,7 +160,7 @@
     "import random\n",
     "import numpy as np\n",
     "import tensorflow as tf\n",
-    "from wandb.keras import WandbMetricsLogger, WandbModelCheckpoint\n",
+    "from wandb.integration.keras import WandbMetricsLogger, WandbModelCheckpoint\n",
     "\n",
     "# Simple Keras Model\n",
     "\n",
@@ -209,7 +209,7 @@
     "    # Add WandbMetricsLogger to log metrics and WandbModelCheckpoint to log model checkpoints\n",
     "    wandb_callbacks = [\n",
     "        WandbMetricsLogger(),\n",
-    "        WandbModelCheckpoint(filepath=\"my_model_{epoch:02d}\"),\n",
+    "        WandbModelCheckpoint(filepath=\"my_model_{epoch:02d}.keras\"),\n",
     "    ]\n",
     "\n",
     "    model.fit(\n",