From 370795c34ccaf7f40c19786aefa8dafc3b436944 Mon Sep 17 00:00:00 2001
From: PythonFZ <fabian.zills@web.de>
Date: Wed, 3 Apr 2024 14:34:39 +0200
Subject: [PATCH] isort

---
 apax/nodes/bal.py   | 8 +++++---
 apax/nodes/md.py    | 2 +-
 apax/nodes/model.py | 6 +++---
 apax/nodes/utils.py | 6 ++++--
 4 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/apax/nodes/bal.py b/apax/nodes/bal.py
index c69d52be..cd2f523b 100644
--- a/apax/nodes/bal.py
+++ b/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.
diff --git a/apax/nodes/md.py b/apax/nodes/md.py
index e35f380d..a1daf35e 100644
--- a/apax/nodes/md.py
+++ b/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__)
 
diff --git a/apax/nodes/model.py b/apax/nodes/model.py
index b9040ecd..f949dda8 100644
--- a/apax/nodes/model.py
+++ b/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
 
diff --git a/apax/nodes/utils.py b/apax/nodes/utils.py
index 3fb363c0..59c6f05d 100644
--- a/apax/nodes/utils.py
+++ b/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.