1c21df6eea15d8da0cd7ea1b82ab24869eb83f48,geomstats/geometry/product_riemannian_metric.py,ProductRiemannianMetric,log,#ProductRiemannianMetric#Any#Any#Any#,238

Before Change

            logs = gs.concatenate(logs, axis=1)
            return logs

        if point_type == "matrix":
            point = gs.to_ndarray(point, to_ndim=2, axis=0)
            point = gs.to_ndarray(point, to_ndim=3, axis=0)
            base_point = gs.to_ndarray(base_point, to_ndim=2, axis=0)
            base_point = gs.to_ndarray(base_point, to_ndim=3, axis=0)
            logs = gs.stack(
                [self.metrics[i].log(point[:, i], base_point[:, i])
                 for i in range(self.n_metrics)], axis=1)
            return logs

        raise ValueError("invalid point_type argument: {}, expected "
                         "either matrix of vector".format(point_type))

After Change

        if point_type is None:
            point_type = self.default_point_type
        geomstats.errors.check_parameter_accepted_values(
            point_type, "point_type", ["vector", "matrix"])

        if point_type == "vector":
            point = gs.to_ndarray(point, to_ndim=2)
            base_point = gs.to_ndarray(base_point, to_ndim=2)
            intrinsic = self.is_intrinsic(base_point)
            args = {"point": point, "base_point": base_point}
            logs = self._iterate_over_metrics("log", args, intrinsic)
            logs = gs.concatenate(logs, axis=1)
            return logs

        point = gs.to_ndarray(point, to_ndim=2, axis=0)
        point = gs.to_ndarray(point, to_ndim=3, axis=0)
        base_point = gs.to_ndarray(base_point, to_ndim=2, axis=0)
        base_point = gs.to_ndarray(base_point, to_ndim=3, axis=0)
        logs = gs.stack(

Instances