BUG: Make n a long int for np.random.multinomial

numpy · Nov 11, 2023 · 12b7b35 · 12b7b35
1 parent 249b5be
commit 12b7b35
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/numpy/random/mtrand.pyx b/numpy/random/mtrand.pyx
@@ -4253,7 +4253,7 @@ cdef class RandomState:
         x.shape = tuple(final_shape)
         return x
 
-    def multinomial(self, np.npy_intp n, object pvals, size=None):
+    def multinomial(self, long n, object pvals, size=None):
         """
         multinomial(n, pvals, size=None)
 

diff --git a/numpy/random/tests/test_randomstate.py b/numpy/random/tests/test_randomstate.py
@@ -183,6 +183,9 @@ def test_multinomial_pvals_float32(self):
         with pytest.raises(ValueError, match=match):
             random.multinomial(1, pvals)
 
+    def test_multinomial_n_float(self):
+        # Non-index integer types should gracefully truncate floats
+        random.multinomial(100.5, [0.2, 0.8])
 
 class TestSetState:
     def setup_method(self):