* Fix Tuple downcast Error (Only Integer)

apache · Aug 8, 2020 · f14bef8 · f14bef8
1 parent ecb4dac
commit f14bef8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/api/operator/numpy/random/np_multinomial_op.cc b/src/api/operator/numpy/random/np_multinomial_op.cc
@@ -49,7 +49,7 @@ MXNET_REGISTER_API("_npi.multinomial")
     inputs[0] = args[1].operator mxnet::NDArray*();
     num_inputs = 1;
   } else {
-    param.pvals = Tuple<double>(args[1].operator ObjectRef());
+    param.pvals = Obj2Tuple<double, Float>(args[1].operator ObjectRef());
   }
 
   // parse size