[TF-DF] Hotfix dtreeviz colab

PiperOrigin-RevId: 666753071
tensorflow · Aug 23, 2024 · 16c057f · 16c057f
1 parent 1e62a79
commit 16c057f
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/documentation/_book.yaml b/documentation/_book.yaml
@@ -71,8 +71,6 @@ upper_tabs:
       - title: Run model with TensorFlow.js
         path: /decision_forests/tf_df_in_tf_js
         status: external
-      - title: Visualizing Trees
-        path: /decision_forests/tutorials/dtreeviz_colab
 
     - name: API Reference
       skip_translation: true

diff --git a/documentation/tutorials/dtreeviz_colab.ipynb b/documentation/tutorials/dtreeviz_colab.ipynb
@@ -120,7 +120,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -q -U tensorflow_decision_forests"
+        "!pip install -q -U tensorflow_decision_forests==1.9.2  # Warning: dtreeviz is not compatible with TF-DF >= 1.10.0"
       ]
     },
     {

diff --git a/documentation/tutorials/index.md b/documentation/tutorials/index.md
@@ -4,8 +4,6 @@ The following notebooks are available:
 
 -   [Beginner Colab](beginner_colab.ipynb): Learn about the basic about model
     training, evaluation and exportation.
--   [Visualizing Trees](dtreeviz_colab.ipynb): How to visualize and
-    interpret decision trees using [dtreeviz](https://github.com/parrt/dtreeviz).
 -   [Ranking Colab](ranking_colab.ipynb): Learn about ranking with decision
     forests.
 -   [Intermediate Colab](intermediate_colab.ipynb): How to consume text and