Skip to content

Commit ea4c9d4

Browse files
authored
Resolve indexing issue when loading medshapenet dataset. (#10472)
1 parent 8172bdb commit ea4c9d4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch_geometric/datasets/medshapenet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,8 @@ def process(self) -> None:
9696

9797
subset = []
9898
for dataset in list_of_datasets:
99-
self.newpath = self.root + '/' + dataset.split("/")[1]
99+
parts = dataset.split("/")
100+
self.newpath = self.root + '/' + parts[1 if len(parts) > 1 else 0]
100101
if not os.path.exists(self.newpath):
101102
os.makedirs(self.newpath)
102103
stl_files = msn_instance.dataset_files(dataset, '.stl')

0 commit comments

Comments
 (0)