diff --git a/torax/tests/persistent_cache_test.py b/torax/tests/persistent_cache_test.py
index e97f8440d..749e899c9 100644
--- a/torax/tests/persistent_cache_test.py
+++ b/torax/tests/persistent_cache_test.py
@@ -22,7 +22,6 @@
 
 from absl.testing import absltest
 from absl.testing import parameterized
-
 import torax
 from torax import run_simulation_main
 
diff --git a/torax/tests/scripts/run_and_save_all_benchmarks.py b/torax/tests/scripts/run_and_save_all_benchmarks.py
index 6b1c842cf..10a1e510c 100644
--- a/torax/tests/scripts/run_and_save_all_benchmarks.py
+++ b/torax/tests/scripts/run_and_save_all_benchmarks.py
@@ -28,7 +28,6 @@
 from torax._src.test_utils import paths
 from torax._src.torax_pydantic import model_config
 
-
 _OUTPUT_DIR = flags.DEFINE_string(
     'output_dir', '/tmp/torax_sim_outputs', 'Where to save sim outputs.'
 )