Commit 8059fa6f authored by Da Zheng's avatar Da Zheng Committed by Quan (Andy) Gan
Browse files

print errors if users sample from mutable graph. (#649)

* give warning.

* fix.

* update error msg.
parent 059b1a6d
...@@ -230,6 +230,8 @@ class NeighborSampler(NodeFlowSampler): ...@@ -230,6 +230,8 @@ class NeighborSampler(NodeFlowSampler):
``len(seed_nodes) // batch_size`` (if ``seed_nodes`` is None, then it is equal ``len(seed_nodes) // batch_size`` (if ``seed_nodes`` is None, then it is equal
to the set of all nodes in the graph). to the set of all nodes in the graph).
Note: NeighborSampler currently only supprts immutable graphs.
Parameters Parameters
---------- ----------
g : DGLGraph g : DGLGraph
...@@ -296,6 +298,8 @@ class NeighborSampler(NodeFlowSampler): ...@@ -296,6 +298,8 @@ class NeighborSampler(NodeFlowSampler):
g, batch_size, seed_nodes, shuffle, num_workers * 2 if prefetch else 0, g, batch_size, seed_nodes, shuffle, num_workers * 2 if prefetch else 0,
ThreadPrefetchingWrapper) ThreadPrefetchingWrapper)
assert g.is_readonly, "NeighborSampler doesn't support mutable graphs. " + \
"Please turn it into an immutable graph with DGLGraph.readonly"
assert node_prob is None, 'non-uniform node probability not supported' assert node_prob is None, 'non-uniform node probability not supported'
assert isinstance(expand_factor, Integral), 'non-int expand_factor not supported' assert isinstance(expand_factor, Integral), 'non-int expand_factor not supported'
...@@ -330,6 +334,8 @@ class LayerSampler(NodeFlowSampler): ...@@ -330,6 +334,8 @@ class LayerSampler(NodeFlowSampler):
The NodeFlow loader returns a list of NodeFlows. The NodeFlow loader returns a list of NodeFlows.
The size of the NodeFlow list is the number of workers. The size of the NodeFlow list is the number of workers.
Note: LayerSampler currently only supprts immutable graphs.
Parameters Parameters
---------- ----------
g : DGLGraph g : DGLGraph
...@@ -380,6 +386,8 @@ class LayerSampler(NodeFlowSampler): ...@@ -380,6 +386,8 @@ class LayerSampler(NodeFlowSampler):
g, batch_size, seed_nodes, shuffle, num_workers * 2 if prefetch else 0, g, batch_size, seed_nodes, shuffle, num_workers * 2 if prefetch else 0,
ThreadPrefetchingWrapper) ThreadPrefetchingWrapper)
assert g.is_readonly, "LayerSampler doesn't support mutable graphs. " + \
"Please turn it into an immutable graph with DGLGraph.readonly"
assert node_prob is None, 'non-uniform node probability not supported' assert node_prob is None, 'non-uniform node probability not supported'
self._num_workers = int(num_workers) self._num_workers = int(num_workers)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment