Fix Notebook tests.

PiperOrigin-RevId: 621553263

Author: fineguy

.github/actions/setup/action.yml

@@ -14,6 +14,10 @@ inputs:
     description: 'Extra dependencies'
     required: false
     default: 'tests-all'
+  use-cache:
+    description: 'Use cache'
+    required: false
+    default: 'true'
 
 runs:
   using: 'composite'
@@ -51,13 +55,18 @@ runs:
     - name: Store inputs for identifying Python cache
       run: echo "${{ env.today }}-${{ inputs.extras }}" >> python-cache-keys.txt
       shell: bash
-    - uses: actions/setup-python@v5
+    - if: ${{ inputs.use-cache == 'true' }}
+      uses: actions/setup-python@v5
       with:
         python-version: ${{ inputs.python-version }}
         cache: 'pip'
         cache-dependency-path: |
           setup.py
           python-cache-keys.txt
+    - if: ${{ inputs.use-cache == 'false' }}
+      uses: actions/setup-python@v5
+      with:
+        python-version: ${{ inputs.python-version }}
 
     # Install Python deps.
     # If Python cache is restored then we don't need to install dependencies again.

.github/workflows/pytest.yml

@@ -185,8 +185,7 @@ jobs:
     - uses: ./.github/actions/setup
       with:
         tf-version: tensorflow
-
-    - run: pip install ipython
+        use-cache: false
 
     # Test each notebook sequentially.
     - name: Run notebook
@@ -202,6 +201,7 @@ jobs:
                 --ExecutePreprocessor.timeout=600 \
                 --ExecutePreprocessor.kernel_name=tfds-notebook \
                 --to notebook \
-                --execute $notebook
+                --execute $notebook && \
+              pip install tensorflow  # reinstall tensorflow if it was uninstalled
             fi
           done

docs/data_source.ipynb

@@ -47,13 +47,13 @@
         "    <a target=\"_blank\" href=\"https://www.tensorflow.org/datasets/tfless_tfds\"><img src=\"https://www.tensorflow.org/images/tf_logo_32px.png\" />View on TensorFlow.org</a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a target=\"_blank\" href=\"https://colab.research.google.com/github/tensorflow/datasets/blob/master/docs/tfless_tfds.ipynb\"><img src=\"https://www.tensorflow.org/images/colab_logo_32px.png\" />Run in Google Colab</a>\n",
+        "    <a target=\"_blank\" href=\"https://colab.research.google.com/github/tensorflow/datasets/blob/master/docs/data_source.ipynb\"><img src=\"https://www.tensorflow.org/images/colab_logo_32px.png\" />Run in Google Colab</a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a target=\"_blank\" href=\"https://github.com/tensorflow/datasets/blob/master/docs/tfless_tfds.ipynb\"><img src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" />View source on GitHub</a>\n",
+        "    <a target=\"_blank\" href=\"https://github.com/tensorflow/datasets/blob/master/docs/data_source.ipynb\"><img src=\"https://www.tensorflow.org/images/GitHub-Mark-32px.png\" />View source on GitHub</a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a href=\"https://storage.googleapis.com/tensorflow_docs/datasets/docs/tfless_tfds.ipynb\"><img src=\"https://www.tensorflow.org/images/download_logo_32px.png\" />Download notebook</a>\n",
+        "    <a href=\"https://storage.googleapis.com/tensorflow_docs/datasets/docs/data_source.ipynb\"><img src=\"https://www.tensorflow.org/images/download_logo_32px.png\" />Download notebook</a>\n",
         "  </td>\n",
         "</table>"
       ]