diff --git a/torch_geometric_temporal/dataset/chickenpox.py b/torch_geometric_temporal/dataset/chickenpox.py index 7801c48f..079d1c6a 100644 --- a/torch_geometric_temporal/dataset/chickenpox.py +++ b/torch_geometric_temporal/dataset/chickenpox.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -19,7 +20,11 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/chickenpox.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + + context = ssl._create_unverified_context() + self._dataset = json.loads( + urllib.request.urlopen(url, context=context).read() + ) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/encovid.py b/torch_geometric_temporal/dataset/encovid.py index f798734f..29dbf6f4 100644 --- a/torch_geometric_temporal/dataset/encovid.py +++ b/torch_geometric_temporal/dataset/encovid.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import DynamicGraphTemporalSignal @@ -21,7 +22,8 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/england_covid.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) def _get_edges(self): self._edges = [] diff --git a/torch_geometric_temporal/dataset/metr_la.py b/torch_geometric_temporal/dataset/metr_la.py index cb732481..00ba68ec 100644 --- a/torch_geometric_temporal/dataset/metr_la.py +++ b/torch_geometric_temporal/dataset/metr_la.py @@ -1,5 +1,6 @@ import os -import urllib +import ssl +import urllib.request import zipfile import numpy as np import torch @@ -25,7 +26,8 @@ def __init__(self, raw_data_dir=os.path.join(os.getcwd(), "data")): self._read_web_data() def _download_url(self, url, save_path): # pragma: no cover - with urllib.request.urlopen(url) as dl_file: + context = ssl._create_unverified_context() + with urllib.request.urlopen(url, context=context) as dl_file: with open(save_path, "wb") as out_file: out_file.write(dl_file.read()) diff --git a/torch_geometric_temporal/dataset/montevideo_bus.py b/torch_geometric_temporal/dataset/montevideo_bus.py index 1ca62c25..3a23aaff 100644 --- a/torch_geometric_temporal/dataset/montevideo_bus.py +++ b/torch_geometric_temporal/dataset/montevideo_bus.py @@ -1,6 +1,7 @@ from typing import List import json -import urllib +import ssl +import urllib.request import numpy as np from torch_geometric_temporal.signal import StaticGraphTemporalSignal @@ -22,7 +23,8 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/montevideo_bus.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) def _get_node_ids(self): return [node.get('bus_stop') for node in self._dataset["nodes"]] diff --git a/torch_geometric_temporal/dataset/mtm.py b/torch_geometric_temporal/dataset/mtm.py index 958112ea..5648ca83 100644 --- a/torch_geometric_temporal/dataset/mtm.py +++ b/torch_geometric_temporal/dataset/mtm.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from torch_geometric_temporal.signal import StaticGraphTemporalSignal @@ -22,7 +23,8 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/mtm_1.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/pedalme.py b/torch_geometric_temporal/dataset/pedalme.py index 4e3eb4c8..6d822364 100644 --- a/torch_geometric_temporal/dataset/pedalme.py +++ b/torch_geometric_temporal/dataset/pedalme.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -18,7 +19,8 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/pedalme_london.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/pems_bay.py b/torch_geometric_temporal/dataset/pems_bay.py index a6c18265..acc05f44 100644 --- a/torch_geometric_temporal/dataset/pems_bay.py +++ b/torch_geometric_temporal/dataset/pems_bay.py @@ -1,5 +1,6 @@ import os -import urllib +import ssl +import urllib.request import zipfile import numpy as np import torch @@ -25,7 +26,8 @@ def __init__(self, raw_data_dir=os.path.join(os.getcwd(), "data")): self._read_web_data() def _download_url(self, url, save_path): # pragma: no cover - with urllib.request.urlopen(url) as dl_file: + context = ssl._create_unverified_context() + with urllib.request.urlopen(url, context=context) as dl_file: with open(save_path, "wb") as out_file: out_file.write(dl_file.read()) diff --git a/torch_geometric_temporal/dataset/twitter_tennis.py b/torch_geometric_temporal/dataset/twitter_tennis.py index 5dd96612..5d511299 100644 --- a/torch_geometric_temporal/dataset/twitter_tennis.py +++ b/torch_geometric_temporal/dataset/twitter_tennis.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import DynamicGraphTemporalSignal @@ -78,7 +79,8 @@ def _read_web_data(self): "https://raw.githubusercontent.com/ferencberes/pytorch_geometric_temporal/developer/dataset/" + fname ) - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) # with open("/home/fberes/git/pytorch_geometric_temporal/dataset/"+fname) as f: # self._dataset = json.load(f) diff --git a/torch_geometric_temporal/dataset/wikimath.py b/torch_geometric_temporal/dataset/wikimath.py index e60d1719..6ec4eb6d 100644 --- a/torch_geometric_temporal/dataset/wikimath.py +++ b/torch_geometric_temporal/dataset/wikimath.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -19,7 +20,8 @@ def __init__(self): def _read_web_data(self): url = "https://raw.githubusercontent.com/benedekrozemberczki/pytorch_geometric_temporal/master/dataset/wikivital_mathematics.json" - self._dataset = json.loads(urllib.request.urlopen(url).read()) + context = ssl._create_unverified_context() + self._dataset = json.loads(urllib.request.urlopen(url, context=context).read()) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/windmilllarge.py b/torch_geometric_temporal/dataset/windmilllarge.py index 2e7e2cdc..261a176f 100644 --- a/torch_geometric_temporal/dataset/windmilllarge.py +++ b/torch_geometric_temporal/dataset/windmilllarge.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -16,7 +17,10 @@ def __init__(self): def _read_web_data(self): url = "https://graphmining.ai/temporal_datasets/windmill_output.json" - self._dataset = json.loads(urllib.request.urlopen(url).read().decode()) + context = ssl._create_unverified_context() + self._dataset = json.loads( + urllib.request.urlopen(url, context=context).read().decode() + ) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/windmillmedium.py b/torch_geometric_temporal/dataset/windmillmedium.py index f33853c1..48f06032 100644 --- a/torch_geometric_temporal/dataset/windmillmedium.py +++ b/torch_geometric_temporal/dataset/windmillmedium.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -16,7 +17,10 @@ def __init__(self): def _read_web_data(self): url = "https://graphmining.ai/temporal_datasets/windmill_output_medium.json" - self._dataset = json.loads(urllib.request.urlopen(url).read().decode()) + context = ssl._create_unverified_context() + self._dataset = json.loads( + urllib.request.urlopen(url, context=context).read().decode() + ) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T diff --git a/torch_geometric_temporal/dataset/windmillsmall.py b/torch_geometric_temporal/dataset/windmillsmall.py index eee2d5ff..d7fb8db3 100644 --- a/torch_geometric_temporal/dataset/windmillsmall.py +++ b/torch_geometric_temporal/dataset/windmillsmall.py @@ -1,5 +1,6 @@ import json -import urllib +import ssl +import urllib.request import numpy as np from ..signal import StaticGraphTemporalSignal @@ -16,7 +17,10 @@ def __init__(self): def _read_web_data(self): url = "https://graphmining.ai/temporal_datasets/windmill_output_small.json" - self._dataset = json.loads(urllib.request.urlopen(url).read().decode()) + context = ssl._create_unverified_context() + self._dataset = json.loads( + urllib.request.urlopen(url, context=context).read().decode() + ) def _get_edges(self): self._edges = np.array(self._dataset["edges"]).T