Skip to content

Commit

Permalink
fix typo in edge definition
Browse files Browse the repository at this point in the history
  • Loading branch information
AMHermansen committed Sep 11, 2023
1 parent f08613d commit 75ee094
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions src/graphnet/models/graphs/graph_definition.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def __init__(

# Member Variables
self._detector = detector
self._edge_definiton = edge_definition
self._edge_definition = edge_definition
self._node_definition = node_definition
if node_feature_names is None:
# Assume all features in Detector is used.
Expand Down Expand Up @@ -113,8 +113,8 @@ def forward( # type: ignore
graph.n_pulses = torch.tensor(len(node_features), dtype=torch.int32)

# Assign edges
if self._edge_definiton is not None:
graph = self._edge_definiton(graph)
if self._edge_definition is not None:
graph = self._edge_definition(graph)
else:
self.warnonce(
"No EdgeDefinition provided. Graphs will not have edges defined!"
Expand Down Expand Up @@ -154,7 +154,6 @@ def forward( # type: ignore
def _validate_input(
self, node_features: np.array, node_feature_names: List[str]
) -> None:

# node feature matrix dimension check
assert node_features.shape[1] == len(node_feature_names)

Expand Down

0 comments on commit 75ee094

Please sign in to comment.