Support button-cells without tfo class (#2304)

tensorflow · Apr 23, 2024 · e520630 · e520630
1 parent 10eef0f
commit e520630
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/tensorflow_docs/tools/nblint/style/tensorflow.py b/tools/tensorflow_docs/tools/nblint/style/tensorflow.py
@@ -81,7 +81,9 @@ 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):