google-deepmind / torax

TORAX: Tokamak transport simulation in JAX
https://torax.readthedocs.io
Other
345 stars 31 forks source link

Ignore extra kwargs in stepper runtime params. #209

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Ignore extra kwargs in stepper runtime params.

Presently all stepper runtime params are flat in config['stepper']. This leads to issues such as raising KeyError when switching a stepper from NewtonRaphson to Linear, and keeping log_iterations in the config. This is not user-friendly since it is beneficial to toggle steppers for testing/comparison and not need to worry about spurious fields from other stepper runtime params.