fix makedirs for python2
@@ -230,7 +230,8 @@ def main():
dataset = D.get_qm9(preprocessor, labels=labels)
# Cache the laded dataset.
- os.makedirs(cache_dir, exist_ok=True)
+ if not os.path.exists(cache_dir):
+ os.makedirs(cache_dir)
NumpyTupleDataset.save(dataset_cache_path, dataset)
# Scale the label values, if necessary.
