Remove dependency on six

Also remove unused "import io"

14 files changed, 12 insertions, 23 deletions

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 3e89686..a28c644 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -8,4 +8,3 @@ https://data.pyg.org/whl/torch-1.10.0%2Bcpu/torch_spline_conv-1.2.1-cp37-cp37m-l
 torch_geometric
 sphinx==4.0.2
 sphinx_rtd_theme==0.5.2
-six
diff --git a/setup.py b/setup.py
index e0f84e2..a2f7f02 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,6 @@ install_requires = [
     "torch_scatter",
     "torch_geometric",
     "numpy",
-    "six",
     "networkx",
 ]
 tests_require = ["pytest", "pytest-cov", "mock", "networkx", "tqdm"]
diff --git a/torch_geometric_temporal/dataset/chickenpox.py b/torch_geometric_temporal/dataset/chickenpox.py
index 15e0533..7801c48 100644
--- a/torch_geometric_temporal/dataset/chickenpox.py
+++ b/torch_geometric_temporal/dataset/chickenpox.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/encovid.py b/torch_geometric_temporal/dataset/encovid.py
index 73043c6..f798734 100644
--- a/torch_geometric_temporal/dataset/encovid.py
+++ b/torch_geometric_temporal/dataset/encovid.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import DynamicGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/metr_la.py b/torch_geometric_temporal/dataset/metr_la.py
index d59bef7..cb73248 100644
--- a/torch_geometric_temporal/dataset/metr_la.py
+++ b/torch_geometric_temporal/dataset/metr_la.py
@@ -1,9 +1,9 @@
 import os
+import urllib
 import zipfile
 import numpy as np
 import torch
 from torch_geometric.utils import dense_to_sparse
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/montevideo_bus.py b/torch_geometric_temporal/dataset/montevideo_bus.py
index 3117c54..1ca62c2 100644
--- a/torch_geometric_temporal/dataset/montevideo_bus.py
+++ b/torch_geometric_temporal/dataset/montevideo_bus.py
@@ -1,7 +1,7 @@
 from typing import List
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from torch_geometric_temporal.signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/mtm.py b/torch_geometric_temporal/dataset/mtm.py
index a2bc11f..958112e 100644
--- a/torch_geometric_temporal/dataset/mtm.py
+++ b/torch_geometric_temporal/dataset/mtm.py
@@ -1,7 +1,6 @@
-from typing import List
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from torch_geometric_temporal.signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/pedalme.py b/torch_geometric_temporal/dataset/pedalme.py
index 8f4a835..4e3eb4c 100644
--- a/torch_geometric_temporal/dataset/pedalme.py
+++ b/torch_geometric_temporal/dataset/pedalme.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/pems_bay.py b/torch_geometric_temporal/dataset/pems_bay.py
index b90c989..a6c1826 100644
--- a/torch_geometric_temporal/dataset/pems_bay.py
+++ b/torch_geometric_temporal/dataset/pems_bay.py
@@ -1,9 +1,9 @@
 import os
+import urllib
 import zipfile
 import numpy as np
 import torch
 from torch_geometric.utils import dense_to_sparse
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/twitter_tennis.py b/torch_geometric_temporal/dataset/twitter_tennis.py
index 64ccbd4..5dd9661 100644
--- a/torch_geometric_temporal/dataset/twitter_tennis.py
+++ b/torch_geometric_temporal/dataset/twitter_tennis.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import DynamicGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/wikimath.py b/torch_geometric_temporal/dataset/wikimath.py
index 9dea930..e60d171 100644
--- a/torch_geometric_temporal/dataset/wikimath.py
+++ b/torch_geometric_temporal/dataset/wikimath.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/windmilllarge.py b/torch_geometric_temporal/dataset/windmilllarge.py
index 66ccfa5..2e7e2cd 100644
--- a/torch_geometric_temporal/dataset/windmilllarge.py
+++ b/torch_geometric_temporal/dataset/windmilllarge.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/windmillmedium.py b/torch_geometric_temporal/dataset/windmillmedium.py
index 06f0152..f33853c 100644
--- a/torch_geometric_temporal/dataset/windmillmedium.py
+++ b/torch_geometric_temporal/dataset/windmillmedium.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal
 
 
diff --git a/torch_geometric_temporal/dataset/windmillsmall.py b/torch_geometric_temporal/dataset/windmillsmall.py
index aaf265e..eee2d5f 100644
--- a/torch_geometric_temporal/dataset/windmillsmall.py
+++ b/torch_geometric_temporal/dataset/windmillsmall.py
@@ -1,7 +1,6 @@
-import io
 import json
+import urllib
 import numpy as np
-from six.moves import urllib
 from ..signal import StaticGraphTemporalSignal