Skip to content

Commit

Permalink
Update python/dgl/graphbolt/impl/gpu_cached_feature.py
Browse files Browse the repository at this point in the history
  • Loading branch information
mfbalin authored Jul 20, 2024
1 parent 5625d3a commit f3f5733
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/dgl/graphbolt/impl/gpu_cached_feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,9 @@ class _Waiter:
@staticmethod
def wait():
"""Returns the stored value when invoked."""
values[missing_index] = missing_values_future.wait()
missing_values = missing_values_future.wait()
self._feature.replace(missing_keys, missing_values)
values[missing_index] = missing_values
return values

yield _Waiter()
Expand Down

0 comments on commit f3f5733

Please sign in to comment.