Update Python in python tests

Summary:
Update python to fix test breaks.

Test Plan:
Tested locally.

.github/workflows/main_python.yml

@@ -17,10 +17,10 @@ jobs:
 
     steps:
     - uses: actions/checkout@v2
-    - name: Set up Python 3.8
+    - name: Set up Python 3.10
       uses: actions/setup-python@v2
       with:
-        python-version: 3.8
+        python-version: '3.10'
     - name: Install PyTorch
       run: |
         python -m pip install --upgrade pip