TypeError: to_networkx() got an unexpected keyword argument 'graph_attrs'

[AdarshMJ]

Hi @rusty1s @EdisonLeeeee

I currently have PyTorch/1.11-CUDA-11.5 and PyTorch-Geometric/2.0.4 installed. When I try convert PyG graphs to networks.. I am getting this error -

 from torch_geometric.utils import from_networkx, to_networkx
Gbefore = to_networkx(data, node_attrs=["x"], graph_attrs=["y"], to_undirected=True)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: to_networkx() got an unexpected keyword argument 'graph_attrs'

Should I change the PyTorch/PyG versions?

[rusty1s]

Yes, the graph_attrs argument was added in PyG 2.1.0, see here.

[AdarshMJ]

Got it. Thank you :)