diff --git a/.gitignore b/.gitignore
index d2278fa..1b49006 100644
--- a/.gitignore
+++ b/.gitignore
@@ -188,4 +188,4 @@ _version.py
 *.code-workspace
 
 /config*
-*.pt
\ No newline at end of file
+*.pt
diff --git a/src/anemoi/graphs/create.py b/src/anemoi/graphs/create.py
index cbbe879..d2b803d 100644
--- a/src/anemoi/graphs/create.py
+++ b/src/anemoi/graphs/create.py
@@ -1,11 +1,11 @@
+import logging
 import os
+
 import torch
 from anemoi.utils.config import DotDict
 from hydra.utils import instantiate
 from torch_geometric.data import HeteroData
 
-import logging
-
 logger = logging.getLogger(__name__)
 
 
diff --git a/src/anemoi/graphs/edges/attributes.py b/src/anemoi/graphs/edges/attributes.py
index 15f6caf..0aa297d 100644
--- a/src/anemoi/graphs/edges/attributes.py
+++ b/src/anemoi/graphs/edges/attributes.py
@@ -1,8 +1,8 @@
+import logging
 from abc import ABC
 from abc import abstractmethod
 from typing import Optional
 
-import logging
 import numpy as np
 from torch_geometric.data import HeteroData