Skip to content

GitLab

Unverified Commit c264a410 authored by Mingbang Wang's avatar Mingbang Wang Committed by GitHub
Browse files

[Misc] rename a variable (#7260)

parent 0bd79a2b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/dgl/graphbolt/impl/ondisk_dataset.py
View file @ c264a410
...@@ -877,16 +877,16 @@ class OnDiskDataset(Dataset): ...@@ -877,16 +877,16 @@ class OnDiskDataset(Dataset):
names=tuple(data.name for data in tvt_set[0].data), names=tuple(data.name for data in tvt_set[0].data),
) )
else: else:
data = {} itemsets = {}
for tvt in tvt_set: for tvt in tvt_set:
data[tvt.type] = ItemSet( itemsets[tvt.type] = ItemSet(
tuple( tuple(
read_data(data.path, data.format, data.in_memory) read_data(data.path, data.format, data.in_memory)
for data in tvt.data for data in tvt.data
), ),
names=tuple(data.name for data in tvt.data), names=tuple(data.name for data in tvt.data),
) )
ret = ItemSetDict(data) ret = ItemSetDict(itemsets)
return ret return ret
def _init_all_nodes_set(self, graph) -> Union[ItemSet, ItemSetDict]: def _init_all_nodes_set(self, graph) -> Union[ItemSet, ItemSetDict]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment