replace long with int for python3 (apache#40)

reminisce · May 19, 2017 · 2881185 · 2881185
1 parent b80f95a
commit 2881185
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/test_utils.py b/python/mxnet/test_utils.py
@@ -74,7 +74,7 @@ def rand_sparse_ndarray(shape, storage_type, density=None):
         # TODO(haibin) support high dim sparse ndarray
         assert(len(shape) < 3)
         prod = np.prod(shape)
-        num_cols = long(prod / shape[0])
+        num_cols = int(prod / shape[0])
         # sample index
         idx_sample = rnd.rand(shape[0])
         indices = np.argwhere(idx_sample < density).flatten()