Update CI to match larq/larq

.github/workflows/unittest.yml

@@ -11,27 +11,31 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.6, 3.7]
-        tf-version: [1.14.0, 1.15.2, 2.0.1, 2.1.0]
+        tf-version: [1.14.0, 1.15.3, 2.0.2, 2.1.1, 2.2.0, 2.3.0]
+        python-version: [3.7]
+        include:
+          - tf-version: 2.2.0
+            python-version: 3.6
+          - tf-version: 2.2.0
+            python-version: 3.8
 
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python
         uses: actions/setup-python@v2.1.1
         with:
-          python-version: ${{matrix.python-version}}
+          python-version: ${{ matrix.python-version }}
       - name: Fix tfds for TF 1.x
         run: pip install tensorflow_datasets==1.3.*
         if: matrix.tf-version == '1.14.0'
       - name: Install dependencies
         run: |
-          pip install tensorflow==${{matrix.tf-version}}
+          pip install tensorflow-cpu==${{matrix.tf-version}} || pip install tensorflow==${{matrix.tf-version}}
           pip install larq
           pip install -e .[test]
       - name: Test with pytest
         run: pytest . -vv --cov=zookeeper --cov-report=xml --cov-config=.coveragerc
       - name: Test example script # Test that the model successfully compiles.
         run: python examples/larq_experiment.py BinaryNetMnist epochs=0 --dataset.download
-      - name: Upload code-coverage report to Codecov
-        if: matrix.python-version == '3.7' && matrix.tf-version == '2.0.1'
-        run: curl -s https://codecov.io/bash | bash -s -- -f ./coverage.xml -F unittests
+      - name: Upload coverage to Codecov
+        run: bash <(curl -s https://codecov.io/bash) -f ./coverage.xml -F unittests