Closed zhang-zhen-project closed 1 week 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).
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
Fixed on latest master.
🐛 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