Merge pull request #28324 from charris/update-test_dirichlet_moderately_small_alpha

charris · web-flow · commit 010ad9b59799 · 2025-02-12T09:28:04.000-07:00
CI: update test_moderately_small_alpha [wheel build]
diff --git a/numpy/random/tests/test_generator_mt19937.py b/numpy/random/tests/test_generator_mt19937.py
@@ -1244,7 +1244,7 @@ def test_dirichlet_small_alpha(self):
     @pytest.mark.slow
     def test_dirichlet_moderately_small_alpha(self):
         # Use alpha.max() < 0.1 to trigger stick breaking code path
-        alpha = np.array([0.02, 0.04, 0.03])
+        alpha = np.array([0.02, 0.04])
         exact_mean = alpha / alpha.sum()
         random = Generator(MT19937(self.seed))
         sample = random.dirichlet(alpha, size=20000000)
