da94c5e70d32738cc17575265c1445904b4b752e,tests/test_dirichlet_distributions.py,TestDirichletDistributions,test_point_to_pdf,#TestDirichletDistributions#,52

Before Change

        points = self.dirichlet.random_uniform(self.n_samples)
        pdfs = self.dirichlet.point_to_pdf(points)
        alpha = gs.ones(self.dim)
        random_point_in_simplex = self.dirichlet.sample(alpha)
        pdfs_at_point = pdfs(random_point_in_simplex)
        self.assertAllClose(gs.shape(pdfs_at_point), (self.n_samples))

    @geomstats.tests.np_and_pytorch_only

After Change

        alpha = gs.ones(self.dim)
        samples = self.dirichlet.sample(alpha, self.n_samples)
        result = pdfs(samples)
        pdf1 = [dirichlet.pdf(x, points[0, :]) for x in samples]
        pdf2 = [dirichlet.pdf(x, points[1, :]) for x in samples]
        expected = gs.stack([gs.array(pdf1), gs.array(pdf2)], axis=0)
        self.assertAllClose(result, expected)

    @geomstats.tests.np_and_pytorch_only
    def test_metric_matrix_vectorization(self):

Instances