Some tests are not running on newer versions of python.

[AlankritVerma01]

This is out of the box -
==================================================== short test summary info =====================================================
FAILED torax/config/tests/build_runtime_params_test.py::RuntimeParamsSliceTest::test_wext_in_dynamic_runtime_params_cannot_be_negative - AssertionError: RuntimeError not raised
FAILED torax/tests/interpolated_param_test.py::InterpolatedParamTest::test_interpolated_param_need_xs_to_be_sorted1 - AssertionError: RuntimeError not raised
FAILED torax/tests/jax_utils_test.py::JaxUtilsTest::test_enable_errors - AssertionError: RuntimeError not raised
FAILED torax/tests/run_simulation_main_test.py::RunSimulationMainTest::test_main_app_cc - AssertionError: 1 is not None
FAILED torax/tests/run_simulation_main_test.py::RunSimulationMainTest::test_main_app_runs - AssertionError: 1 is not None
FAILED torax/transport_model/tests/transport_model_test.py::TransportSmoothingTest::test_smoothing - TypeError: Value 'fake' for discriminator 'transport_model' mapped to multiple choices
FAILED torax/transport_model/tests/transport_model_test.py::TransportSmoothingTest::test_smoothing_everywhere - TypeError: Value 'fake' for discriminator 'transport_model' mapped to multiple choices
7 failed, 1384 passed, 10 skipped, 25373 warnings in 384.72s (0:06:24)