* Fix tuple of float

apache · Apr 21, 2020 · 9b68cb4 · 9b68cb4
1 parent 7dd1b8f
commit 9b68cb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/api/operator/numpy/random/np_multinomial_op.cc b/src/api/operator/numpy/random/np_multinomial_op.cc
@@ -44,12 +44,12 @@ 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());