Skip to content

Commit

Permalink
Merge branch 'master' into akihironitta-patch-1
Browse files Browse the repository at this point in the history
  • Loading branch information
akihironitta authored Nov 14, 2023
2 parents 4af590b + b1f8535 commit 171e494
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).

### Changed

- Update `DistNeighborSampler` for homogeneous graphs ([#8209](https://github.com/pyg-team/pytorch_geometric/pull/8209))
- Update `DistNeighborSampler` for homogeneous graphs ([#8209](https://github.com/pyg-team/pytorch_geometric/pull/8209), [#8367](https://github.com/pyg-team/pytorch_geometric/pull/8367))
- Update `GraphStore` and `FeatureStore` to support distributed training ([#8083](https://github.com/pyg-team/pytorch_geometric/pull/8083))
- Disallow the usage of `add_self_loops=True` in `GCNConv(normalize=False)` ([#8210](https://github.com/pyg-team/pytorch_geometric/pull/8210))
- Disable device asserts during `torch_geometric.compile` ([#8220](https://github.com/pyg-team/pytorch_geometric/pull/8220))
Expand Down
3 changes: 2 additions & 1 deletion torch_geometric/sampler/neighbor_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,8 @@ def __init__(
# Obtain graph metadata:
node_attrs = [
attr for attr in feature_store.get_all_tensor_attrs()
if isinstance(attr.group_name, NodeType)
if isinstance(attr.group_name, NodeType) # Heterogeneous ...
or attr.group_name is None # ... or homogeneous.
]
self.node_types = list(set(attr.group_name for attr in node_attrs))

Expand Down

0 comments on commit 171e494

Please sign in to comment.