Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Jun 5, 2023
1 parent b6dc886 commit 29fdffa
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion torch_geometric/datasets/hm.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,8 @@ class HM(InMemoryDataset):
Args:
root (str): Root directory where the dataset should be saved.
use_all_tables_as_node_types (bool, optional): If set to :obj:`True`,
will keep the transaction table as node type. (default: :obj:`False`)
will use the transaction table as a distinct node type.
(default: :obj:`False`)
transform (callable, optional): A function/transform that takes in an
:obj:`torch_geometric.data.HeteroData` object and returns a
transformed version. The data object will be transformed before
Expand Down

0 comments on commit 29fdffa

Please sign in to comment.