Skip to content

GitLab

Unverified Commit a8b27b47 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

disable batching blocks (#2034)

parent a9c4df67
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python/dgl/batch.py
View file @ a8b27b47
...@@ -161,6 +161,8 @@ def batch(graphs, ndata=ALL, edata=ALL, *, node_attrs=None, edge_attrs=None): ...@@ -161,6 +161,8 @@ def batch(graphs, ndata=ALL, edata=ALL, *, node_attrs=None, edge_attrs=None):
if not (is_all(edata) or isinstance(edata, list)): if not (is_all(edata) or isinstance(edata, list)):
raise DGLError('Invalid argument edata: must be a string list but got {}.'.format( raise DGLError('Invalid argument edata: must be a string list but got {}.'.format(
type(edata))) type(edata)))
if any(g.is_block for g in graphs):
raise DGLError("Batching a block is not supported.")
utils.check_all_same_device(graphs, 'graphs') utils.check_all_same_device(graphs, 'graphs')
utils.check_all_same_idtype(graphs, 'graphs') utils.check_all_same_idtype(graphs, 'graphs')
......
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