Update intro_to_modules.ipynb

Added keras import on top of the code.
tensorflow · Oct 17, 2023 · a3c2132 · a3c2132
1 parent a7d2c63
commit a3c2132
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/site/en/guide/intro_to_modules.ipynb b/site/en/guide/intro_to_modules.ipynb
@@ -91,6 +91,7 @@
       "outputs": [],
       "source": [
         "import tensorflow as tf\n",
+        "import keras\n",
         "from datetime import datetime\n",
         "\n",
         "%load_ext tensorboard"
@@ -918,7 +919,6 @@
       },
       "outputs": [],
       "source": [
-        "import keras\n",
         "@keras.saving.register_keras_serializable()\n",
         "class MySequentialModel(tf.keras.Model):\n",
         "  def __init__(self, name=None, **kwargs):\n",