Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Xiang Song committed Jun 17, 2024
1 parent 7cd7176 commit baa9ff7
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions python/graphstorm/inference/mt_infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -203,6 +203,10 @@ def gen_embs(edge_mask=None):
task_infos = recon_nfeat_test_loader.task_infos

with th.no_grad():
# Note: We need to be careful here as embs is a mutable
# object (a dictionary), use a mutable value
# as default value for an argument is danerous.
# pylint: disable=dangerous-default-value
def nfrecon_gen_embs(skip_last_self_loop=False, node_embs=embs):
""" Generate node embeddings for node feature reconstruction
"""
Expand Down

0 comments on commit baa9ff7

Please sign in to comment.