diff --git a/colabs/keras/Use_WandbEvalCallback_in_your_Keras_workflow.ipynb b/colabs/keras/Use_WandbEvalCallback_in_your_Keras_workflow.ipynb index 4cbce176..cd0fac6e 100644 --- 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 index 3c3ddec2..b2c04057 100644 --- 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 index 3ca4c90c..9a3b8384 100644 --- 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" ] }, {