'DGLHeteroGraph' object has no attribute 'dsttype'

when running graphsage

python train_sampling.py
Finished data loading.
NumNodes: 232965
NumEdges: 114848857
NumFeats: 602
NumClasses: 41
NumTrainingSamples: 153431
NumValidationSamples: 23831
NumTestSamples: 55703
Traceback (most recent call last):
File “train_sampling.py”, line 262, in
run(args, device, data)
File “train_sampling.py”, line 202, in run
batch_pred = model(blocks, batch_inputs)
File “/opt/conda/envs/pytorch-ci/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 532, in call
result = self.forward(*input, **kwargs)
File “train_sampling.py”, line 67, in forward
h_dst = h[:block.number_of_nodes(block.dsttype)]
AttributeError: ‘DGLHeteroGraph’ object has no attribute ‘dsttype’


pip install --pre dgl-cu101

Would fix in https://github.com/dmlc/dgl/pull/1371