[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

graphs/tests/edges/test_cutoff.py

@@ -8,9 +8,9 @@
 # nor does it submit to any jurisdiction.
 
 import pytest
+from torch_geometric.data import HeteroData
 
 from anemoi.graphs.edges import CutOffEdges
-from torch_geometric.data import HeteroData
 
 
 def test_init():

graphs/tests/test_normaliser.py

@@ -7,9 +7,8 @@
 # granted to it by virtue of its status as an intergovernmental organisation
 # nor does it submit to any jurisdiction.
 
-import numpy as np
-import torch
 import pytest
+import torch
 
 from anemoi.graphs.normalise import NormaliserMixin