search

google-deepmind / torax

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

Enable shortcuts in `TimeInterpolatedArray`. #198

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Enable shortcuts in TimeInterpolatedArray.

Allow passing in a single float value or single dict value for TimeInterpolatedArray fields.