diff --git a/src/graphnet/models/graphs/nodes/nodes.py b/src/graphnet/models/graphs/nodes/nodes.py
index afe0dcfce..b98ad375a 100644
--- a/src/graphnet/models/graphs/nodes/nodes.py
+++ b/src/graphnet/models/graphs/nodes/nodes.py
@@ -4,9 +4,7 @@
 from abc import abstractmethod
 
 import torch
-from torch_geometric.nn import knn_graph, radius_graph
 from torch_geometric.data import Data
-import numpy as np
 
 from graphnet.utilities.decorators import final
 from graphnet.utilities.config import save_model_config