Skip to content

Commit f76731e

Browse files
deprecate datasets until we find a solution
1 parent 041ee01 commit f76731e

3 files changed

Lines changed: 23 additions & 20 deletions

File tree

test/dataset_test.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -345,26 +345,26 @@ def test_windmilllarge():
345345
assert snapshot.y.shape == (319,)
346346

347347

348-
def test_windmillsmall():
349-
loader = WindmillOutputSmallDatasetLoader()
350-
dataset = loader.get_dataset()
351-
for epoch in range(2):
352-
for snapshot in dataset:
353-
assert snapshot.edge_index.shape == (2, 121)
354-
assert snapshot.edge_attr.shape == (121,)
355-
assert snapshot.x.shape == (11, 8)
356-
assert snapshot.y.shape == (11,)
357-
358-
359-
def test_windmillmedium():
360-
loader = WindmillOutputMediumDatasetLoader()
361-
dataset = loader.get_dataset()
362-
for epoch in range(2):
363-
for snapshot in dataset:
364-
assert snapshot.edge_index.shape == (2, 676)
365-
assert snapshot.edge_attr.shape == (676,)
366-
assert snapshot.x.shape == (26, 8)
367-
assert snapshot.y.shape == (26,)
348+
# def test_windmillsmall():
349+
# loader = WindmillOutputSmallDatasetLoader()
350+
# dataset = loader.get_dataset()
351+
# for epoch in range(2):
352+
# for snapshot in dataset:
353+
# assert snapshot.edge_index.shape == (2, 121)
354+
# assert snapshot.edge_attr.shape == (121,)
355+
# assert snapshot.x.shape == (11, 8)
356+
# assert snapshot.y.shape == (11,)
357+
358+
359+
# def test_windmillmedium():
360+
# loader = WindmillOutputMediumDatasetLoader()
361+
# dataset = loader.get_dataset()
362+
# for epoch in range(2):
363+
# for snapshot in dataset:
364+
# assert snapshot.edge_index.shape == (2, 676)
365+
# assert snapshot.edge_attr.shape == (676,)
366+
# assert snapshot.x.shape == (26, 8)
367+
# assert snapshot.y.shape == (26,)
368368

369369

370370
def test_covid():

torch_geometric_temporal/dataset/windmillmedium.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ class WindmillOutputMediumDatasetLoader(object):
1313
"""
1414

1515
def __init__(self):
16+
raise RuntimeError("The WindmillMedium dataset is no longer accessible via 'graphmining.ai'. If you have a local copy, please contact the PGT maintainers, and we will re-upload the dataset. ")
1617
self._read_web_data()
1718

1819
def _read_web_data(self):

torch_geometric_temporal/dataset/windmillsmall.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@ class WindmillOutputSmallDatasetLoader(object):
1313
"""
1414

1515
def __init__(self):
16+
raise RuntimeError("The WindmillSmall dataset is no longer accessible via 'graphmining.ai'. If you have a local copy, please contact the PGT maintainers, and we will re-upload the dataset. ")
17+
1618
self._read_web_data()
1719

1820
def _read_web_data(self):

0 commit comments

Comments
 (0)