From 7f96dc6e75b24b062c20f29a7e81d2be7a7e6ad3 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 5 Feb 2025 17:08:31 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 graphs/src/anemoi/graphs/edges/attributes.py | 2 +-
 graphs/tests/edges/test_edge_attributes.py   | 4 +---
 2 files changed, 2 insertions(+), 4 deletions(-)

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")