Update Python in python tests

Summary:
Update python to fix test breaks.

Test Plan:
Tested locally.

Author: c-p-i-o

.github/workflows/main_python.yml

@@ -16,11 +16,11 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v2
-    - name: Set up Python 3.8
-      uses: actions/setup-python@v2
+    - uses: actions/checkout@v4
+    - name: Set up Python 3.10
+      uses: actions/setup-python@v5
       with:
-        python-version: 3.8
+        python-version: '3.10'
     - name: Install PyTorch
       run: |
         python -m pip install --upgrade pip