keras import (#532)

wandb · Jun 14, 2024 · 3903016 · 3903016
1 parent 9c27c3b
commit 3903016
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/colabs/keras/Use_WandbEvalCallback_in_your_Keras_workflow.ipynb b/colabs/keras/Use_WandbEvalCallback_in_your_Keras_workflow.ipynb
@@ -81,9 +81,9 @@
     "\n",
     "# Weights and Biases related imports\n",
     "import wandb\n",
-    "from wandb.keras import WandbMetricsLogger\n",
-    "from wandb.keras import WandbModelCheckpoint\n",
-    "from wandb.keras import WandbEvalCallback"
+    "from wandb.integration.keras import WandbMetricsLogger\n",
+    "from wandb.integration.keras import WandbModelCheckpoint\n",
+    "from wandb.integration.keras import WandbEvalCallback"
    ]
   },
   {

diff --git a/colabs/keras/Use_WandbMetricLogger_in_your_Keras_workflow.ipynb b/colabs/keras/Use_WandbMetricLogger_in_your_Keras_workflow.ipynb
@@ -81,7 +81,7 @@
     "\n",
     "# Weights and Biases related imports\n",
     "import wandb\n",
-    "from wandb.keras import WandbMetricsLogger"
+    "from wandb.integration.keras import WandbMetricsLogger"
    ]
   },
   {

diff --git a/colabs/keras/Use_WandbModelCheckpoint_in_your_Keras_workflow.ipynb b/colabs/keras/Use_WandbModelCheckpoint_in_your_Keras_workflow.ipynb
@@ -80,8 +80,8 @@
     "\n",
     "# Weights and Biases related imports\n",
     "import wandb\n",
-    "from wandb.keras import WandbMetricsLogger\n",
-    "from wandb.keras import WandbModelCheckpoint"
+    "from wandb.integration.keras import WandbMetricsLogger\n",
+    "from wandb.integration.keras import WandbModelCheckpoint"
    ]
   },
   {