diff --git a/doc/source/examples/cd_text_imdb.ipynb b/doc/source/examples/cd_text_imdb.ipynb index 0a17c312a..c4a924c28 100644 --- 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 index 0a17c312a..c4a924c28 100644 --- 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",