Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
mananshah99 committed Jun 28, 2022
1 parent bb410c8 commit 1bf110b
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions torch_geometric/data/hetero_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -680,9 +680,7 @@ def _get_tensor_size(self, attr: TensorAttr) -> Tuple:
def get_all_tensor_attrs(self) -> List[TensorAttr]:
out = []
for group_name, group in self.node_items():
print(group_name, group)
for attr_name in group:
print(attr_name)
if group.is_node_attr(attr_name):
out.append(TensorAttr(group_name, attr_name))
return out
Expand Down

0 comments on commit 1bf110b

Please sign in to comment.