google-deepmind / torax

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

Move transport coefficient clipping to TransportModel class. #208

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Move transport coefficient clipping to TransportModel class.

Removes redundancy in child classes. Tests with constant transport model and pedestal changed with minor impact (~1e-2 relative errors in stationary state), since the constant transport model did not previously carry out chi and D clipping in the pedestal zone.