diff --git a/graphs/src/anemoi/graphs/edges/attributes.py b/graphs/src/anemoi/graphs/edges/attributes.py
index 9b69c801..e5343242 100644
--- a/graphs/src/anemoi/graphs/edges/attributes.py
+++ b/graphs/src/anemoi/graphs/edges/attributes.py
@@ -10,7 +10,7 @@
 from __future__ import annotations
 
 import logging
-from abc import ABC, ABCMeta
+from abc import ABC
 
 import torch
 from torch_geometric.data.storage import NodeStorage
diff --git a/graphs/tests/edges/test_edge_attributes.py b/graphs/tests/edges/test_edge_attributes.py
index e30b5cf5..955f3e2e 100644
--- a/graphs/tests/edges/test_edge_attributes.py
+++ b/graphs/tests/edges/test_edge_attributes.py
@@ -8,18 +8,16 @@
 # nor does it submit to any jurisdiction.
 
 
-from typing import TYPE_CHECKING
 
 import pytest
 import torch
+from torch_geometric.data import HeteroData
 
 from anemoi.graphs.edges.attributes import AttributeFromSourceNode
 from anemoi.graphs.edges.attributes import AttributeFromTargetNode
 from anemoi.graphs.edges.attributes import EdgeDirection
 from anemoi.graphs.edges.attributes import EdgeLength
 
-from torch_geometric.data import HeteroData
-
 TEST_EDGES = ("test_nodes", "to", "test_nodes")