Skip to content

Commit bd06e8b

Browse files
committed
threw DirichletDistribution into one of the test_concrete_models.py tests just for good measure
1 parent 6b7733e commit bd06e8b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/qinfer/tests/test_concrete_models.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
NormalDistribution,
5555
BetaDistribution, UniformDistribution,
5656
PostselectedDistribution,
57-
ConstrainedSumDistribution,
57+
ConstrainedSumDistribution, DirichletDistribution,
5858
DirectViewParallelizedModel
5959
)
6060
from qinfer.ale import ALEApproximateModel
@@ -268,8 +268,7 @@ class TestMultinomialModel(ConcreteModelTest, DerandomizedTestCase):
268268
def instantiate_model(self):
269269
return MultinomialModel(NDieModel(n=6))
270270
def instantiate_prior(self):
271-
unif = UniformDistribution(np.array([[0,1],[0,1],[0,1],[0,1],[0,1],[0,1]]))
272-
return ConstrainedSumDistribution(unif, desired_total=1)
271+
return DirichletDistribution([1,2,3,10,1,3])
273272
def instantiate_expparams(self):
274273
return np.arange(10).astype(self.model.expparams_dtype)
275274

0 commit comments

Comments
 (0)