isort

apax/nodes/bal.py

@@ -1,12 +1,14 @@
-import zntrack
 import ase
-import numpy as np
 import matplotlib.pyplot as plt
-from .utils import get_flat_data_from_dict, plot_with_uncertainty
+import numpy as np
 import znflow
+import zntrack
 
 from apax.bal import kernel_selection
 
+from .utils import get_flat_data_from_dict, plot_with_uncertainty
+
+
 class BatchKernelSelection(zntrack.Node):
     """Interface to the batch active learning methods implemented in apax.
     Check the apax documentation for a list and explanation of implemented properties.

apax/nodes/md.py

@@ -10,8 +10,8 @@
 import zntrack.utils
 from zntrack import dvc, zn
 
-from .utils import check_duplicate_keys
 from .model import Apax
+from .utils import check_duplicate_keys
 
 log = logging.getLogger(__name__)
 

apax/nodes/model.py

@@ -9,13 +9,13 @@
 import pandas as pd
 import yaml
 import zntrack.utils
+from jax import config
+from matplotlib import pyplot as plt
+from zntrack import dvc, zn
 
 from apax.md import ASECalculator
 from apax.md.function_transformations import available_transformations
 from apax.train.run import run as apax_run
-from jax import config
-from matplotlib import pyplot as plt
-from zntrack import dvc, zn
 
 from .utils import check_duplicate_keys
 

apax/nodes/utils.py

@@ -1,7 +1,9 @@
-import numpy as np
-import matplotlib.pyplot as plt
 import logging
 
+import matplotlib.pyplot as plt
+import numpy as np
+
+
 def get_flat_data_from_dict(data: dict, silent_ignore: bool = False) -> list:
     """Flatten a dictionary of lists into a single list.