Use ubuntu-22.04 across Github Actions.

PiperOrigin-RevId: 708231695

Author: fineguy

.github/workflows/pytest-optional.yml

@@ -18,7 +18,7 @@ concurrency:
 jobs:
   activate-tests:
     name: Check whether we should run tests or not
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
 
     steps:
     - uses: actions/checkout@v3
@@ -36,10 +36,10 @@ jobs:
       fail-fast: false
       matrix:
         tf-version: [tensorflow, tf-nightly]
-        os-version: [ubuntu-latest, macos-latest]
+        os-version: [ubuntu-22.04, macos-latest]
         exclude:
           - tf-version: tensorflow
-            os-version: ubuntu-latest
+            os-version: ubuntu-22.04
 
     name: 'Core TFDS tests'
     uses: ./.github/workflows/pytest-template.yml

.github/workflows/pytest-template.yml

@@ -14,7 +14,7 @@ env:
 jobs:
   shards-job:
     name: Generate shards
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
 
     steps:
       - name: Create variables

.github/workflows/pytest.yml

@@ -32,7 +32,7 @@ concurrency:
 jobs:
   activate-tests:
     name: Check whether we should run tests or not
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
 
     steps:
     - uses: actions/checkout@v3
@@ -50,7 +50,7 @@ jobs:
     uses: ./.github/workflows/pytest-template.yml
     with:
       tf-version: tensorflow
-      os-version: ubuntu-latest
+      os-version: ubuntu-22.04
 
   huggingface-pytest-job:
     needs: activate-tests
@@ -59,7 +59,7 @@ jobs:
     # HuggingFace tests need to be run separately because they're disabled without installed
     # `datasets` library.
     name: 'HuggingFace Python 3.10 tests'
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     timeout-minutes: 30
 
     steps:
@@ -81,7 +81,7 @@ jobs:
     if: ${{ needs.activate-tests.outputs.status }}
 
     name: 'Github API tests'
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     timeout-minutes: 30
 
     steps:
@@ -100,7 +100,7 @@ jobs:
     if: ${{ needs.activate-tests.outputs.status }}
 
     name: 'Notebook tests'
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     timeout-minutes: 30
 
     steps:

.github/workflows/release-nightly.yml

@@ -17,7 +17,7 @@ jobs:
   publish-job:
     # Prevents action from running on forks.
     if: github.repository == 'tensorflow/datasets'
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
 
     steps:
     - name: Set nightly build variables

.github/workflows/release.yml

@@ -19,7 +19,7 @@ jobs:
   publish-job:
     # Prevents action from running on forks and non-master branches.
     if: github.repository == 'tensorflow/datasets'
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     permissions:
       contents: write