Enable CI to use Python 3.10

Currently we are using `python-nightly` in GitHub actions but don't block on test failures, and some tests are failing. We should change this to use Python 3.10 and fix any tests that are still failing. 

The relevant file is `.github/workflows/test.yml`.

Part of #10201.