Skip to content

Commit 58aef82

Browse files
Fix downloaded file check
1 parent 09b1525 commit 58aef82

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

torch_geometric_temporal/dataset/metr_la.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,11 +37,9 @@ def __init__(self, raw_data_dir=os.path.join(os.getcwd(), "data"), index: bool =
3737

3838
def _download_url(self, url, save_path): # pragma: no cover
3939
# Check if file is in data folder from working directory, otherwise download
40-
if not os.path.isfile(
41-
os.path.join(self.raw_data_dir,save_path)
42-
):
40+
if not os.path.isfile(save_path):
4341
print("Downloading to", save_path, flush=True)
44-
42+
4543
response = requests.get(url, stream=True)
4644
file_size = int(response.headers.get('content-length', 0))
4745

0 commit comments

Comments
 (0)