pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.54k stars 3.57k forks source link

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

Closed zhang-zhen-project closed 1 week ago

zhang-zhen-project commented 1 month ago

🐛 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 commented 1 month ago

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 commented 1 week ago

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 commented 1 week ago

Fixed on latest master.