Remove accidentally modified files (linter)

torax/transport_model/qlknn_wrapper.py

@@ -26,13 +26,11 @@
 import functools
 import logging
 import os
-from typing import Callable
-from typing import Final
+from typing import Callable, Final
 
 import chex
 import jax
 from jax import numpy as jnp
-
 from torax import geometry
 from torax import state
 from torax.config import runtime_params_slice
@@ -42,6 +40,7 @@
 from torax.transport_model import runtime_params as runtime_params_lib
 from torax.transport_model import transport_model
 
+
 # Environment variable for the QLKNN model. Used if the model path
 # is not set in the config.
 MODEL_PATH_ENV_VAR: Final[str] = 'TORAX_QLKNN_MODEL_PATH'

torax/transport_model/qualikiz_wrapper.py

@@ -21,18 +21,17 @@
 
 from __future__ import annotations
 
+from collections.abc import Callable
 import dataclasses
 import datetime
 import os
 import subprocess
 import tempfile
-from collections.abc import Callable
 
 import chex
 import numpy as np
 from qualikiz_tools.qualikiz_io import inputfiles as qualikiz_inputtools
 from qualikiz_tools.qualikiz_io import qualikizrun as qualikiz_runtools
-
 from torax import geometry
 from torax import jax_utils
 from torax import state