Skip to content

Commit

Permalink
Add asserts to avoid linting failure
Browse files Browse the repository at this point in the history
  • Loading branch information
michailramp committed Dec 9, 2024
1 parent 2f2bc78 commit 5f6c41d
Showing 1 changed file with 3 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,9 @@ def __init__(self, d_p: int, use_laplacian: bool = True):
self._use_laplacian = use_laplacian

def forward(self, data: Data) -> Data:
assert data.num_nodes is not None
assert data.edge_index is not None

n = data.num_nodes
if self._use_laplacian is True:
node_ids = self._get_lap_eigenvectors(data.edge_index, n)
Expand Down

0 comments on commit 5f6c41d

Please sign in to comment.