update

pyg-team · Jun 13, 2023 · 5573540 · 5573540
1 parent 92be21d
commit 5573540
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 ### Added
 
+- Allow GPU tensors as input to `NodeLoader` and `LinkLoader` ([#7572](https://github.com/pyg-team/pytorch_geometric/pull/7572))
 - Added an `embedding_device` option to allow for GPU inference in `BasicGNN` ([#7548](https://github.com/pyg-team/pytorch_geometric/pull/7548))
 - Added `Performer` to `GPSConv` and remove `attn_dropout` argument from `GPSConv` ([#7465](https://github.com/pyg-team/pytorch_geometric/pull/7465))
 - Enabled `LinkNeighborLoader` to return number of sampled nodes and edges per hop ([#7516](https://github.com/pyg-team/pytorch_geometric/pull/7516))

diff --git a/torch_geometric/sampler/base.py b/torch_geometric/sampler/base.py
@@ -94,7 +94,7 @@ def __getitem__(self, index: Union[Tensor, Any]) -> 'NodeSamplerInput':
         )
 
 
-@dataclass
+@dataclass(init=False)
 class EdgeSamplerInput(CastMixin):
     r"""The sampling input of
     :meth:`~torch_geometric.sampler.BaseSampler.sample_from_edges`.