Change GitHub Actions to ensure works on latest Python minor version.

Author: asmith26

.github/workflows/pythontest.yml

@@ -13,27 +13,23 @@ jobs:
   test:
 
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: [3.7, 3.8]
 
     steps:
     - name: Checkout jax_toolkit repo
       uses: actions/checkout@v2
       with:
         lfs: False
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v1
+    - name: Set up Python 3.x
+      uses: actions/setup-python@v2
       with:
-        python-version: ${{ matrix.python-version }}
+        python-version: '3.x'
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install --no-cache-dir -r requirements-dev.txt
-        pip install .[losses_utils]
+        make pip-install
     - name: Run lint
       run: |
         make lint
     - name: Run tests
       run: |
-        make coverage
+        make coverage

Makefile

@@ -47,11 +47,11 @@ pip-compile:                                 ## Compile requirements.txt from se
 
 pip-install:                                 ## Install dependencies into current environment
 	pip install -r requirements-dev.txt
+	pip install -e .\[losses_utils\]
 
 pip-sync:                                    ## Update your conda/virtual environment to reflect exactly (this will install/upgrade/uninstall everything necessary) what's in requirements.txt
 	pip-sync
 
 pip-sync-dev: pip-sync pip-install           ## Update your conda/virtual environment to reflect exactly what's in requirements.txt, and then install the dev requirements.
-	pip install .\[losses_utils\]
 
 test-all: clean lint coverage-with-report    ## Run all checks