01673d1a6dcb41a20e19f951ee450c44c07aeafd,geomstats/riemannian_metric.py,RiemannianMetric,mean,#RiemannianMetric#Any#Any#Any#Any#Any#,242

Before Change

            variance = self.variance(points=points,
                                     weights=weights,
                                     base_point=mean_next)
            if gs.isclose(variance, 0.)[0, 0]:
                break
            if (sq_dist <= epsilon * variance)[0, 0]:
                break

After Change

                gs.less_equal(sq_dist, epsilon * variance))
            return result[0, 0]

        last_iteration, mean, variance, sq_dist = gs.while_loop(
            lambda i, m, v, sq: while_loop_cond(i, m, v, sq),
            lambda i, m, v, sq: while_loop_body(i, m, v, sq),
            loop_vars=[iteration, mean, variance, sq_dist],
            maximum_iterations=n_max_iterations)
        //while iteration < n_max_iterations:

        //    if gs.isclose(variance, 0.)[0, 0]:
        //        break

Instances