Merge pull request #633 from AMHermansen/fix-knngraph

cleaning code
graphnet-team · Nov 24, 2023 · d5dca17 · d5dca17
2 parents 3b9adc7 + 0fe168c
commit d5dca17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/graphnet/models/graphs/graphs.py b/src/graphnet/models/graphs/graphs.py
@@ -16,7 +16,7 @@ class KNNGraph(GraphDefinition):
     def __init__(
         self,
         detector: Detector,
-        node_definition: NodeDefinition = NodesAsPulses(),
+        node_definition: NodeDefinition = None,
         input_feature_names: Optional[List[str]] = None,
         dtype: Optional[torch.dtype] = torch.float,
         perturbation_dict: Optional[Dict[str, float]] = None,
@@ -44,7 +44,7 @@ def __init__(
         # Base class constructor
         super().__init__(
             detector=detector,
-            node_definition=node_definition,
+            node_definition=node_definition or NodesAsPulses(),
             edge_definition=KNNEdges(
                 nb_nearest_neighbours=nb_nearest_neighbours,
                 columns=columns,