Modify plotruns to take a Mapping of plot data.#2126

copybara-service[bot] merged 1 commit into

maingoogle-deepmind/torax:mainfrom

test_901325655google-deepmind/torax:test_901325655Copy head branch name to clipboard