2b4d328be9d22522eca9a91d39eb4c13c3afafe3,geomstats/learning/frechet_mean.py,,variance,#Any#Any#Any#Any#Any#,19

Before Change

    weights : array-like, shape=[n_samples, 1], optional
    
    n_points = 1
    if point_type == "vector" and gs.ndim(points) == 2:
        n_points = gs.shape(points)[0]
    if point_type == "matrix" and gs.ndim(points) == 3:
        n_points = gs.shape(points)[0]

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

    sum_weights = gs.sum(weights)

After Change

    weights : array-like, shape=[n_samples, 1], optional
    
    n_points = geomstats.vectorization.get_n_points(
        points, point_type)

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

Instances