From 897e69eac9bc7752c0b4259ca0898d1d9dfa9dbd Mon Sep 17 00:00:00 2001
From: Ke Han <kehan5800@gmail.com>
Date: Tue, 21 Apr 2020 00:32:36 +0800
Subject: [PATCH] * Fix tuple of float

---
 src/api/operator/numpy/random/np_multinomial_op.cc | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/src/api/operator/numpy/random/np_multinomial_op.cc b/src/api/operator/numpy/random/np_multinomial_op.cc
index 3df4165bdca4..e3fc375daff9 100644
--- a/src/api/operator/numpy/random/np_multinomial_op.cc
+++ b/src/api/operator/numpy/random/np_multinomial_op.cc
@@ -44,18 +44,14 @@ MXNET_REGISTER_API("_npi.multinomial")
   if (args[1].type_code() == kNull) {
     param.pvals = dmlc::nullopt;
   } else {
-    param.pvals = Tuple<double>(args[1].operator ObjectRef());
+    param.pvals = Obj2Tuple<double, Float>(args[1].operator ObjectRef());
   }
 
   // parse size
   if (args[2].type_code() == kNull) {
-    param.size = dmlc::nullopt;
+    param.size = dmlc::optional<mxnet::Tuple<int>>();
   } else {
-    if (args[2].type_code() == kDLInt) {
-      param.size = Tuple<int>(1, args[2].operator int64_t());
-    } else {
-      param.size = Tuple<int>(args[2].operator ObjectRef());
-    }
+    param.size = mxnet::Tuple<int>(args[2].operator ObjectRef());
   }
 
   attrs.parsed = std::move(param);