diff --git a/CHANGELOG.md b/CHANGELOG.md index 3aacfdc34270..c54a41c005f2 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,7 +7,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/). ### Added -- Added `OnDiskDataset` interface with data loader support ([#8066](https://github.com/pyg-team/pytorch_geometric/pull/8066), [#8088](https://github.com/pyg-team/pytorch_geometric/pull/8088)) +- Added `OnDiskDataset` interface with data loader support ([#8066](https://github.com/pyg-team/pytorch_geometric/pull/8066), [#8088](https://github.com/pyg-team/pytorch_geometric/pull/8088), [#8092](https://github.com/pyg-team/pytorch_geometric/pull/8092)) - Added a tutorial for `Node2Vec` and `MetaPath2Vec` usage ([#7938](https://github.com/pyg-team/pytorch_geometric/pull/7938) - Added a tutorial for multi-GPU training with pure PyTorch ([#7894](https://github.com/pyg-team/pytorch_geometric/pull/7894) - Added `edge_attr` support to `ResGatedGraphConv` ([#8048](https://github.com/pyg-team/pytorch_geometric/pull/8048)) diff --git a/test/loader/test_dataloader.py b/test/loader/test_dataloader.py index 2e86ce7b6472..458a91acff02 100644 --- a/test/loader/test_dataloader.py +++ b/test/loader/test_dataloader.py @@ -66,13 +66,14 @@ def test_dataloader(num_workers, device): assert batch.edge_index_batch.tolist() == [0, 0, 0, 0, 1, 1, 1, 1] -def test_dataloader_on_disk_dataset(tmp_path): +@pytest.mark.parametrize('num_workers', num_workers_list) +def test_dataloader_on_disk_dataset(tmp_path, num_workers): dataset = OnDiskDataset(tmp_path) data1 = Data(x=torch.randn(3, 8)) data2 = Data(x=torch.randn(4, 8)) dataset.extend([data1, data2]) - loader = DataLoader(dataset, batch_size=2) + loader = DataLoader(dataset, batch_size=2, num_workers=num_workers) assert len(loader) == 1 batch = next(iter(loader)) assert batch.num_nodes == 7