Merge pull request #2301 from tensorflow:update-video-classification

PiperOrigin-RevId: 629485009
tensorflow · Apr 30, 2024 · 3eac15b · 3eac15b
2 parents 880385b + d6b06f5
commit 3eac15b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/site/en/tutorials/video/video_classification.ipynb b/site/en/tutorials/video/video_classification.ipynb
@@ -84,9 +84,7 @@
         "## Setup\n",
         "\n",
         "Begin by installing and importing some necessary libraries, including:\n",
-        "[remotezip](https://github.com/gtsystem/python-remotezip) to inspect the contents of a ZIP file, [tqdm](https://github.com/tqdm/tqdm) to use a progress bar, [OpenCV](https://opencv.org/) to process video files, [einops](https://github.com/arogozhnikov/einops/tree/master/docs) for performing more complex tensor operations, and [`tensorflow_docs`](https://github.com/tensorflow/docs/tree/master/tools/tensorflow_docs) for embedding data in a Jupyter notebook.\n",
-        "\n",
-        "**Note**: Use TensorFlow 2.10 to run this tutorial. Versions above TensorFlow 2.10 may not run successfully."
+        "[remotezip](https://github.com/gtsystem/python-remotezip) to inspect the contents of a ZIP file, [tqdm](https://github.com/tqdm/tqdm) to use a progress bar, [OpenCV](https://opencv.org/) to process video files, [einops](https://github.com/arogozhnikov/einops/tree/master/docs) for performing more complex tensor operations, and [`tensorflow_docs`](https://github.com/tensorflow/docs/tree/master/tools/tensorflow_docs) for embedding data in a Jupyter notebook."
       ]
     },
     {
@@ -98,8 +96,7 @@
       "outputs": [],
       "source": [
         "!pip install remotezip tqdm opencv-python einops \n",
-        "# Install TensorFlow 2.10\n",
-        "!pip install tensorflow==2.10.0"
+        "!pip install -U tensorflow keras"
       ]
     },
     {

diff --git a/tools/tensorflow_docs/tools/nblint/style/tensorflow.py b/tools/tensorflow_docs/tools/nblint/style/tensorflow.py
@@ -81,7 +81,11 @@ def not_translation(args):
 
 # Button checks
 
-is_button_cell_re = re.compile(r"class.*tfo-notebook-buttons")
+# Look for class="tfo-notebook-buttons" (CSS used on website versions) or the
+# run-in-colab logo (for notebooks that stick to GitHub/Colab).
+is_button_cell_re = re.compile(
+    r"class.*tfo-notebook-buttons|colab_logo_32px\.png"
+)
 
 
 def get_arg_or_fail(user_args, arg_name, arg_fmt):