Fixed cd_text_imdb.ipynb

SeldonIO · Dec 10, 2024 · 9bf3cb4 · 9bf3cb4
1 parent bc81fc0
commit 9bf3cb4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/doc/source/examples/cd_text_imdb.ipynb b/doc/source/examples/cd_text_imdb.ipynb
@@ -51,15 +51,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {
     "id": "qG5BL45LxfoG"
    },
    "outputs": [],
    "source": [
+    "import os\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\"\n",
+    "\n",
     "import nlp\n",
     "import numpy as np\n",
-    "import os\n",
     "import tensorflow as tf\n",
     "from transformers import AutoTokenizer\n",
     "from alibi_detect.cd import KSDrift, MMDDrift\n",

diff --git a/docs-gb/examples/cd_text_imdb.ipynb b/docs-gb/examples/cd_text_imdb.ipynb
@@ -51,15 +51,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {
     "id": "qG5BL45LxfoG"
    },
    "outputs": [],
    "source": [
+    "import os\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\"\n",
+    "\n",
     "import nlp\n",
     "import numpy as np\n",
-    "import os\n",
     "import tensorflow as tf\n",
     "from transformers import AutoTokenizer\n",
     "from alibi_detect.cd import KSDrift, MMDDrift\n",