Fix TF import (#535)

* keras import * quick fix * import
wandb · Jun 18, 2024 · 58cb855 · 58cb855
1 parent e6c3855
commit 58cb855
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb b/colabs/tensorflow/Simple_TensorFlow_Integration.ipynb
@@ -136,7 +136,7 @@
    "outputs": [],
    "source": [
     "import wandb\n",
-    "from wandb.keras import WandbCallback\n"
+    "from wandb.integration.keras import WandbCallback"
    ]
   },
   {