still fixing example

niklasmei · niklasmei · commit 3ff1abfe0766 · 2026-01-23T14:35:28.000+01:00
diff --git a/examples/pretraining_maskpred_example.py b/examples/pretraining_maskpred_example.py
@@ -12,7 +12,6 @@
 import torch
 
 from graphnet.models.detector.prometheus import Prometheus
-from graphnet.models.graphs import KNNGraph
 from graphnet.models.graphs.nodes import NodesAsPulses
 
 
