f371e5207d3719b257979729803d0fe5eae0e9cb,geomstats/riemannian_metric.py,RiemannianMetric,mean,#RiemannianMetric#Any#Any#Any#Any#,219

Before Change

        // TODO(nina): profile this code to study performance,
        // i.e. what to do with sq_dists_between_iterates.

        n_points = len(points)
        assert n_points > 0

        if weights is None:
            weights = gs.ones(n_points)

        n_weights = len(weights)
        assert n_points == n_weights
        sum_weights = gs.sum(weights)

        mean = points[0]

After Change

        // TODO(nina): profile this code to study performance,
        // i.e. what to do with sq_dists_between_iterates.

        points = gs.array(points)
        n_points = gs.shape(points)[0]
        assert n_points > 0

        if weights is None:
            weights = gs.ones(n_points)

        weights = gs.array(weights)
        n_weights = gs.shape(weights)[0]
        assert n_points == n_weights
        sum_weights = gs.sum(weights)

        mean = points[0]

Instances