Avoid warning in Mixture test

Author: ricardoV94

pymc/tests/distributions/test_mixture.py

@@ -1009,12 +1009,12 @@ def setup_class(cls):
     @pytest.mark.parametrize("batch_shape", [(3, 4), (20,)], ids=str)
     def test_with_multinomial(self, batch_shape):
         p = np.random.uniform(size=(*batch_shape, self.mixture_comps, 3))
+        p /= p.sum(axis=-1, keepdims=True)
         n = 100 * np.ones((*batch_shape, 1))
         w = np.ones(self.mixture_comps) / self.mixture_comps
         mixture_axis = len(batch_shape)
         with Model() as model:
-            with pytest.warns(UserWarning, match="parameters sum up to"):
-                comp_dists = Multinomial.dist(p=p, n=n, shape=(*batch_shape, self.mixture_comps, 3))
+            comp_dists = Multinomial.dist(p=p, n=n, shape=(*batch_shape, self.mixture_comps, 3))
             mixture = Mixture(
                 "mixture",
                 w=w,