From 503b1810edcc70c73e99c53066e9d72ce1298e1b Mon Sep 17 00:00:00 2001 From: rusty1s Date: Tue, 16 Aug 2022 12:25:16 +0000 Subject: [PATCH] add node_type_subgraph and edge_type_subgraph to changelog --- CHANGELOG.md | 2 +- test/transforms/test_random_link_split.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index cd085e4bdf1c..39e04dd4c985 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -61,7 +61,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/). - Confirm that `to_hetero()` works with custom functions, *e.g.*, `dropout_adj` ([4653](https://github.com/pyg-team/pytorch_geometric/pull/4653)) - Added the `MLP.plain_last=False` option ([4652](https://github.com/pyg-team/pytorch_geometric/pull/4652)) - Added a check in `HeteroConv` and `to_hetero()` to ensure that `MessagePassing.add_self_loops` is disabled ([4647](https://github.com/pyg-team/pytorch_geometric/pull/4647)) -- Added `HeteroData.subgraph()` support ([#4635](https://github.com/pyg-team/pytorch_geometric/pull/4635)) +- Added `HeteroData.subgraph()`, `HeteroData.node_type_subgraph()` and `HeteroData.edge_type_subgraph()` support ([#4635](https://github.com/pyg-team/pytorch_geometric/pull/4635)) - Added the `AQSOL` dataset ([#4626](https://github.com/pyg-team/pytorch_geometric/pull/4626)) - Added `HeteroData.node_items()` and `HeteroData.edge_items()` functionality ([#4644](https://github.com/pyg-team/pytorch_geometric/pull/4644)) - Added PyTorch Lightning support in GraphGym ([#4511](https://github.com/pyg-team/pytorch_geometric/pull/4511), [#4516](https://github.com/pyg-team/pytorch_geometric/pull/4516) [#4531](https://github.com/pyg-team/pytorch_geometric/pull/4531), [#4689](https://github.com/pyg-team/pytorch_geometric/pull/4689), [#4843](https://github.com/pyg-team/pytorch_geometric/pull/4843)) diff --git a/test/transforms/test_random_link_split.py b/test/transforms/test_random_link_split.py index 95ac9126abc5..032abe081af3 100644 --- a/test/transforms/test_random_link_split.py +++ b/test/transforms/test_random_link_split.py @@ -100,10 +100,8 @@ def test_random_link_split_on_hetero_data(): data['a', 'p'].edge_attr = torch.arange(1500, 2500) transform = RandomLinkSplit(num_val=0.2, num_test=0.2, is_undirected=True, - edge_types=data.edge_types, - rev_edge_types=data.edge_types) + edge_types=('p', 'p')) train_data, val_data, test_data = transform(data) - print(train_data) assert len(train_data['p']) == 1 assert len(train_data['a']) == 1