ci: switch to amazon runners

Author: d4l3k

.github/workflows/unittest.yaml

@@ -11,15 +11,15 @@ jobs:
     strategy:
       matrix:
         os:
-          - ubuntu-20.04
+          - "linux.2xlarge"
           - "linux.4xlarge.nvidia.gpu"
     runs-on: ${{ matrix.os }}
     steps:
-      - name: Setup Python
-        uses: actions/setup-python@v2
+      # This hash has been reviewed for security issues. If upgrading please
+      # rereview the action source.
+      - uses: kishaningithub/setup-python-amazon-linux@46d057bc1c866dd51b4d2ad0c7631e9e8eed13e1
         with:
-          python-version: "3.10"
-          architecture: x64
+          python-version: '3.10'
       - name: Checkout
         uses: actions/checkout@v2
       - name: Install Dependencies