We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
medshapenet
1 parent 8172bdb commit ea4c9d4Copy full SHA for ea4c9d4
torch_geometric/datasets/medshapenet.py
@@ -96,7 +96,8 @@ def process(self) -> None:
96
97
subset = []
98
for dataset in list_of_datasets:
99
- self.newpath = self.root + '/' + dataset.split("/")[1]
+ parts = dataset.split("/")
100
+ self.newpath = self.root + '/' + parts[1 if len(parts) > 1 else 0]
101
if not os.path.exists(self.newpath):
102
os.makedirs(self.newpath)
103
stl_files = msn_instance.dataset_files(dataset, '.stl')
0 commit comments