[Fix] Add RangePartitionBook to __init__ (dmlc#1738)

* update * update
lili0710432 · Jul 2, 2020 · 781b9ce · 781b9ce
1 parent da16ebf
commit 781b9ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/distributed/__init__.py b/python/dgl/distributed/__init__.py
@@ -2,7 +2,7 @@
 
 from .dist_graph import DistGraphServer, DistGraph, node_split, edge_split
 from .partition import partition_graph, load_partition, load_partition_book
-from .graph_partition_book import GraphPartitionBook, PartitionPolicy
+from .graph_partition_book import GraphPartitionBook, RangePartitionBook, PartitionPolicy
 
 from .rpc import *
 from .rpc_server import start_server