[Docker]Update MXNet CI Image (#582)

* use mxnet 20190528 build * fix mxnet api name

[Docker]Update MXNet CI Image (#582)
* use mxnet 20190528 build * fix mxnet api name
ada3786c · Lingfan Yu · Da Zheng · bd2d2c75 · ada3786c · ada3786c
Commit ada3786c authored May 28, 2019 by Lingfan Yu Committed by Da Zheng May 28, 2019
3 changed files
--- a/docker/install/ubuntu_install_mxnet_cpu.sh
+++ b/docker/install/ubuntu_install_mxnet_cpu.sh
-pip3 install mxnet==1.4.1
+pip3 install mxnet==1.5.0b20190528
--- a/docker/install/ubuntu_install_mxnet_gpu.sh
+++ b/docker/install/ubuntu_install_mxnet_gpu.sh
-pip3 install mxnet-cu90==1.4.1
+pip3 install mxnet-cu90==1.5.0b20190528
--- a/python/dgl/backend/mxnet/tensor.py
+++ b/python/dgl/backend/mxnet/tensor.py
@@ -11,7 +11,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,