Install tf-datasets differently for different tf versions in tests CI

andreped · May 9, 2023 · cb348c2 · cb348c2
1 parent a5300c8
commit cb348c2
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -60,10 +60,16 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install dependencies
+        run: pip install wheel setuptools flake8 pytest-cov
+
+      - name: Install tensorflow-datasets
         run: |
-          pip install wheel setuptools flake8 pytest-cov
-          pip install tensorflow==${{ matrix.tf-version }} "tensorflow-datasets<=4.8.2"
-          pip install "protobuf<=3.20" --force-reinstall
+          if [[ ${{ matrix.tf-version }} == 2.12 ]]; then
+            pip install tensorflow-datasets --upgrade
+          else
+            pip install tensorflow==${{ matrix.tf-version }} "tensorflow-datasets<=4.8.2"
+            pip install "protobuf<=3.20" --force-reinstall
+          fi
 
       - name: Download artifact
         uses: actions/download-artifact@master