Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added num_graphs option to SBMDataset #8648

Merged
merged 5 commits into from
Dec 21, 2023
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update
  • Loading branch information
rusty1s committed Dec 21, 2023
commit 4b668c5b84bb1f395a66cfa0caabb4f0201658ad
2 changes: 1 addition & 1 deletion torch_geometric/datasets/sbm_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,9 +61,9 @@ def __init__(

assert num_graphs > 0

self.num_graphs = num_graphs
self.block_sizes = block_sizes
self.edge_probs = edge_probs
self.num_graphs = num_graphs
self.num_channels = num_channels
self.is_undirected = is_undirected

Expand Down
Loading