Prefer local package imports in examples and clarify TF env errors

Author: shenweichen

examples/alias.py

@@ -1,6 +1,13 @@
+import sys
+from pathlib import Path
+
 import matplotlib.pyplot as plt
 import numpy as np
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from ge.alias import alias_sample, create_alias_table
 
 

examples/deepwalk_wiki.py

@@ -1,15 +1,19 @@
 from pathlib import Path
+import sys
 
 import matplotlib.pyplot as plt
 import networkx as nx
 import numpy as np
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from ge import DeepWalk
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 WIKI_GRAPH_PATH = PROJECT_ROOT / "data" / "wiki" / "Wiki_edgelist.txt"
 WIKI_LABEL_PATH = PROJECT_ROOT / "data" / "wiki" / "wiki_labels.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"

examples/line_wiki.py

@@ -1,15 +1,25 @@
 from pathlib import Path
+import sys
 
 import matplotlib.pyplot as plt
 import networkx as nx
 import numpy as np
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
-from ge import LINE
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
+try:
+    from ge import LINE
+except ImportError as exc:
+    raise ImportError(
+        "Unable to import LINE. Use a supported Python/TensorFlow environment "
+        "(for example Python 3.10-3.12 with tensorflow installed)."
+    ) from exc
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 WIKI_GRAPH_PATH = PROJECT_ROOT / "data" / "wiki" / "Wiki_edgelist.txt"
 WIKI_LABEL_PATH = PROJECT_ROOT / "data" / "wiki" / "wiki_labels.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"

examples/node2vec_flight.py

@@ -1,15 +1,19 @@
 from pathlib import Path
+import sys
 
 import matplotlib.pyplot as plt
 import networkx as nx
 import numpy as np
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from ge import Node2Vec
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 FLIGHT_GRAPH_PATH = PROJECT_ROOT / "data" / "flight" / "brazil-airports.edgelist"
 FLIGHT_LABEL_PATH = PROJECT_ROOT / "data" / "flight" / "labels-brazil-airports.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"

examples/node2vec_wiki.py

@@ -1,15 +1,19 @@
 from pathlib import Path
+import sys
 
 import matplotlib.pyplot as plt
 import networkx as nx
 import numpy as np
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from ge import Node2Vec
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 WIKI_GRAPH_PATH = PROJECT_ROOT / "data" / "wiki" / "Wiki_edgelist.txt"
 WIKI_LABEL_PATH = PROJECT_ROOT / "data" / "wiki" / "wiki_labels.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"

examples/sdne_wiki.py

@@ -1,15 +1,25 @@
 from pathlib import Path
+import sys
 
 import matplotlib.pyplot as plt
 import networkx as nx
 import numpy as np
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
-from ge import SDNE
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
+try:
+    from ge import SDNE
+except ImportError as exc:
+    raise ImportError(
+        "Unable to import SDNE. Use a supported Python/TensorFlow environment "
+        "(for example Python 3.10-3.12 with tensorflow installed)."
+    ) from exc
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 WIKI_GRAPH_PATH = PROJECT_ROOT / "data" / "wiki" / "Wiki_edgelist.txt"
 WIKI_LABEL_PATH = PROJECT_ROOT / "data" / "wiki" / "wiki_labels.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"

examples/struc2vec_flight.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+import sys
 import tempfile
 
 import matplotlib.pyplot as plt
@@ -7,10 +8,13 @@
 from sklearn.linear_model import LogisticRegression
 from sklearn.manifold import TSNE
 
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
 from ge import Struc2Vec
 from ge.classify import Classifier, read_node_label
 
-PROJECT_ROOT = Path(__file__).resolve().parents[1]
 FLIGHT_GRAPH_PATH = PROJECT_ROOT / "data" / "flight" / "brazil-airports.edgelist"
 FLIGHT_LABEL_PATH = PROJECT_ROOT / "data" / "flight" / "labels-brazil-airports.txt"
 SMOKE_GRAPH_PATH = PROJECT_ROOT / "tests" / "Wiki_edgelist.txt"