[Dist] import dgl.distributed for examples (#7761)

examples/distributed/graphsage/node_classification.py

@@ -3,6 +3,7 @@ import socket
 import time
 
 import dgl
+import dgl.distributed
 import dgl.nn.pytorch as dglnn
 import numpy as np
 import torch as th

examples/distributed/graphsage/node_classification_unsupervised.py

@@ -3,6 +3,7 @@ import time
 from contextlib import contextmanager
 
 import dgl
+import dgl.distributed
 import dgl.function as fn
 import dgl.nn.pytorch as dglnn
 

examples/distributed/rgcn/node_classification.py

@@ -19,6 +19,7 @@ os.environ["DGLBACKEND"] = "pytorch"
 from functools import partial
 
 import dgl
+import dgl.distributed
 import torch as th
 import torch.multiprocessing as mp
 import torch.nn as nn

examples/pytorch/graphsage/dist/train_dist.py

@@ -4,6 +4,7 @@ import time
 from contextlib import contextmanager
 
 import dgl
+import dgl.distributed
 import dgl.nn.pytorch as dglnn
 
 import numpy as np

examples/pytorch/graphsage/dist/train_dist_transductive.py

@@ -2,7 +2,7 @@ import argparse
 import time
 
 import dgl
-
+import dgl.distributed
 import numpy as np
 import torch as th
 import torch.nn as nn

examples/pytorch/graphsage/dist/train_dist_unsupervised.py

@@ -3,6 +3,7 @@ import time
 from contextlib import contextmanager
 
 import dgl
+import dgl.distributed
 import dgl.function as fn
 import dgl.nn.pytorch as dglnn
 

examples/pytorch/graphsage/dist/train_dist_unsupervised_transductive.py

@@ -2,7 +2,7 @@ import argparse
 import time
 
 import dgl
-
+import dgl.distributed
 import numpy as np
 import torch as th
 import torch.nn.functional as F

examples/pytorch/rgcn/experimental/entity_classify_dist.py

@@ -19,6 +19,7 @@ os.environ["DGLBACKEND"] = "pytorch"
 from functools import partial
 
 import dgl
+import dgl.distributed
 import torch as th
 import torch.multiprocessing as mp
 import torch.nn as nn