test against 2.0.1 stable pytorch version (#793)

Summary:
Pull Request resolved: #793

Adding unit test for stable pytorch version (2.0.1). This will prevent future changes from breaking OSS compatibility

bypass-github-export-checks

Reviewed By: galrotem

Differential Revision: D56442435

fbshipit-source-id: afa1f0e33f18c5330fcdb7fba7b569df06f2e1d1

Author: JKSenthil

.github/workflows/test.yaml

@@ -6,7 +6,7 @@ on:
   pull_request:
 
 jobs:
-  unit_tests:
+  unit_tests_nightly_pytorch:
     runs-on: ubuntu-latest
     strategy:
       matrix:
@@ -37,3 +37,31 @@ jobs:
           pytest --cov=. --cov-report xml tests -vv
       - name: Upload Coverage to Codecov
         uses: codecov/codecov-action@v2
+  unit_tests_stable_pytorch:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: [3.8]
+    steps:
+      - name: Check out repo
+        uses: actions/checkout@v2
+      - name: Setup conda env
+        uses: conda-incubator/setup-miniconda@v2
+        with:
+          miniconda-version: "latest"
+          activate-environment: test
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        shell: bash -l {0}
+        run: |
+          set -eux
+          conda activate test
+          pip install -r requirements.txt
+          pip install -r dev-requirements.txt
+          pip install  --no-build-isolation -e .
+      - name: Run unit tests with coverage
+        shell: bash -l {0}
+        run: |
+          set -eux
+          conda activate test
+          pytest --cov=. --cov-report xml tests -vv