Remove unnecessary extra creation of StateHistory.

PiperOrigin-RevId: 669319133
google-deepmind · Aug 30, 2024 · 5047477 · 5047477
1 parent 1873c51
commit 5047477
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torax/simulation_app.py b/torax/simulation_app.py
@@ -301,7 +301,8 @@ def main(
   output_file = write_simulation_output_to_file(output_dir, ds)
 
   if log_sim_output:
-    history = output.StateHistory(torax_outputs)
-    log_simulation_output_to_stdout(history.core_profiles, geo, history.times)
+    log_simulation_output_to_stdout(
+        state_history.core_profiles, geo, state_history.times
+    )
 
   return ds, output_file