Skip to content

Commit 5b3090c

Browse files
fix new loader to create dir if it does not exist yet
1 parent f76731e commit 5b3090c

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

test/dataset_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ def test_wiki():
335335

336336

337337
def test_windmilllarge():
338-
loader = WindmillOutputLargeDatasetLoader()
338+
loader = WindmillOutputLargeDatasetLoader(raw_data_dir="/tmp/")
339339
dataset = loader.get_dataset()
340340
for epoch in range(2):
341341
for snapshot in dataset:

torch_geometric_temporal/dataset/windmilllarge.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def __init__(self, raw_data_dir=os.path.join(os.getcwd(), "data"), index=False):
3030

3131

3232
def _read_web_data(self):
33+
os.makedirs(self.raw_data_dir, exist_ok=True)
3334
if not os.path.isfile(
3435
os.path.join(self.raw_data_dir, "windmill_output.json")
3536
):

0 commit comments

Comments
 (0)