Removing the warning that appears in the Graph Transformer example (#6061)

83437e67 · Andrei Ivanov · GitHub · ec7137dd · 83437e67
Unverified Commit 83437e67 authored Aug 02, 2023 by Andrei Ivanov Committed by GitHub Aug 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/sparse/graph_transformer.py examples/sparse/graph_transformer.py +1 -1

No files found.
--- a/examples/sparse/graph_transformer.py
+++ b/examples/sparse/graph_transformer.py
@@ -181,7 +181,7 @@ if __name__ == "__main__":
    evaluator = Evaluator("ogbg-molhiv")
    # laplacian positional encoding
    for g, _ in tqdm(dataset, desc="Computing Laplacian PE"):
-        g.ndata["PE"] = dgl.laplacian_pe(g, k=pos_enc_size, padding=True)
+        g.ndata["PE"] = dgl.lap_pe(g, k=pos_enc_size, padding=True)

    # Create model.
    out_size = dataset.num_tasks