[Docker]Move to the latest MXNet version. (#610)

* use the right version of mxnet. * fix.

[Docker]Move to the latest MXNet version. (#610)
* use the right version of mxnet. * fix.
70ee8664 · Da Zheng · GitHub · 653428bd · 70ee8664 · 70ee8664
Unverified Commit 70ee8664 authored Jun 06, 2019 by Da Zheng Committed by GitHub Jun 06, 2019
3 changed files
--- a/docker/install/ubuntu_install_mxnet_cpu.sh
+++ b/docker/install/ubuntu_install_mxnet_cpu.sh
-pip3 install mxnet==1.5.0b20190523
+pip3 install mxnet==1.5.0b20190605
--- a/docker/install/ubuntu_install_mxnet_gpu.sh
+++ b/docker/install/ubuntu_install_mxnet_gpu.sh
-pip3 install mxnet-cu90==1.5.0b20190523
+pip3 install mxnet-cu90==1.5.0b20190605
--- a/python/dgl/backend/mxnet/tensor.py
+++ b/python/dgl/backend/mxnet/tensor.py
@@ -13,7 +13,7 @@ MX_VERSION = LooseVersion(mx.__version__)
 # After MXNet 1.5, empty tensors aren't supprted by default.
 # after we turn on the numpy compatible flag, MXNet supports empty NDArray.
 if MX_VERSION.version[0] == 1 and MX_VERSION.version[1] >= 5:
-    mx.set_np_compat(True)
+    mx.set_np_shape(True)
 def data_type_dict():
    return {'float16' : np.float16,