6f4671eb1b466bf5c484dea3830d4e1bb61709b8,geomstats/invariant_metric.py,InvariantMetric,__init__,#InvariantMetric#Any#Any#Any#,24

Before Change

        if inner_product_mat_at_identity.ndim == 3:
            n_mats, _, _ = inner_product_mat_at_identity.shape
            assert n_mats == 1
            inner_product_mat_at_identity = gs.squeeze(
                               inner_product_mat_at_identity, axis=0)

        matrix_shape = inner_product_mat_at_identity.shape
        assert matrix_shape == (group.dimension,) * 2
        assert left_or_right in ("left", "right")

After Change

        inner_product_mat_at_identity = gs.to_ndarray(
            inner_product_mat_at_identity, to_ndim=3)
        mat_shape = inner_product_mat_at_identity.shape
        assert mat_shape == (1,) + (group.dimension, ) * 2, mat_shape

        assert left_or_right in ("left", "right")

        eigenvalues = gs.linalg.eigvalsh(inner_product_mat_at_identity)

Instances