Skip to content

Commit 78bbfbd

Browse files
Hotfix: Remove duplicate node_types in NeighborLoader with FeatureStore (#4856)
* fix * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 7a89399 commit 78bbfbd

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

torch_geometric/data/feature_store.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,8 @@ def get_tensor(self, *args, **kwargs) -> FeatureTensorType:
329329
return self._to_type(attr, tensor)
330330

331331
def _multi_get_tensor(
332-
self, attrs: List[TensorAttr]) -> Optional[FeatureTensorType]:
332+
self,
333+
attrs: List[TensorAttr]) -> List[Optional[FeatureTensorType]]:
333334
r"""To be implemented by :class:`FeatureStore` subclasses.
334335
335336
.. note::

torch_geometric/loader/neighbor_loader.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -108,9 +108,8 @@ def __init__(
108108
node_attrs = feature_store.get_all_tensor_attrs()
109109
edge_attrs = graph_store.get_all_edge_attrs()
110110

111-
self.node_types = [
112-
node_attr.group_name for node_attr in node_attrs
113-
]
111+
self.node_types = list(
112+
set(node_attr.group_name for node_attr in node_attrs))
114113
self.edge_types = [edge_attr.edge_type for edge_attr in edge_attrs]
115114

116115
# Set other required parameters:

0 commit comments

Comments
 (0)