Internal change.

PiperOrigin-RevId: 794058519

Author: sbodenstein

torax/run_simulation_main.py

@@ -23,7 +23,6 @@
 from collections.abc import Sequence
 import enum
 import functools
-import os
 import pathlib
 import time
 
@@ -37,6 +36,7 @@
 from torax._src.plotting import plotruns_lib
 from torax._src.torax_pydantic import model_config
 
+
 # String used when prompting the user to make a choice of command
 CHOICE_PROMPT = 'Your choice: '
 # String used when prompting the user to make a yes / no choice