Quick start The first example failed with "dataset = TUDataset(root='/home/zhangzhen/pygstart/dataset/ENZYMES', name='ENZYMES')"

[zhang-zhen-project]

🐛 Describe the bug

import torch
from torch_geometric.data import Data
from torch_geometric.datasets import TUDataset

dataset = TUDataset(root='/home/zhangzhen/pygstart/dataset/ENZYMES', name='ENZYMES')

Versions

[pip3] numpy==1.24.4
[pip3] torch==2.1.0+cu121.with.pypi.cudnn
[pip3] torch-cluster==1.6.3+pt21cu121
[pip3] torch_geometric==2.5.3
[pip3] torch-scatter==2.1.2+pt21cu121
[pip3] torch-sparse==0.6.18+pt21cu121
[pip3] torch-spline-conv==1.2.2+pt21cu121
[pip3] triton==2.1.0
[conda] numpy 1.24.4 pypi_0 pypi
[conda] torch 2.1.0+cu121.with.pypi.cudnn pypi_0 pypi
[conda] torch-cluster 1.6.3+pt21cu121 pypi_0 pypi
[conda] torch-geometric 2.5.3 pypi_0 pypi
[conda] torch-scatter 2.1.2+pt21cu121 pypi_0 pypi
[conda] torch-sparse 0.6.18+pt21cu121 pypi_0 pypi
[conda] torch-spline-conv 1.2.2+pt21cu121 pypi_0 pypi
[conda] triton 2.1.0 pypi_0 pypi

[rusty1s]

What is the error message? I think we fixed this already on master (there was an issue with loading on TUDataset on Windows due to an upgrade in fsspec dependency).

[natalia-dubljevic]

I'm also running into issues loading TUDataset. I'm using WSL2, and version 2.5.2 of pyg. The full error message is:

---

TypeError Traceback (most recent call last)
Cell In[22], line 1
----> 1 dataset = TUDataset(root="./data", name='ENZYMES')

File /home/dubljevi/miniconda/envs/deep_env/lib/python3.10/site-packages/torch_geometric/datasets/tu_dataset.py:129, in TUDataset.init(self, root, name, transform, pre_transform, pre_filter, force_reload, use_node_attr, use_edge_attr, cleaned)
127 self.name = name
128 self.cleaned = cleaned
--> 129 super().init(root, transform, pre_transform, pre_filter,
130 force_reload=force_reload)
132 out = fs.torch_load(self.processed_paths[0])
133 if not isinstance(out, tuple) or len(out) < 3:

File /home/dubljevi/miniconda/envs/deep_env/lib/python3.10/site-packages/torch_geometric/data/in_memory_dataset.py:81, in InMemoryDataset.init(self, root, transform, pre_transform, pre_filter, log, force_reload)
72 def init(
73 self,
74 root: Optional[str] = None,
(...)
79 force_reload: bool = False,
80 ) -> None:
---> 81 super().init(root, transform, pre_transform, pre_filter, log,
82 force_reload)
84 self._data: Optional[BaseData] = None
85 self.slices: Optional[Dict[str, Tensor]] = None
...
191 fs2 = get_fs(path2)
192 assert fs1.protocol == fs2.protocol
--> 193 fs1.mv(path1, path2, recursive)

TypeError: LocalFileSystem.mv() takes 3 positional arguments but 4 were given

[rusty1s]

Fixed on latest master.