f5813466fb9d202def50f783c0fdeeb164da50d6,geomstats/visualization.py,,plot,#Any#Any#Any#,133

Before Change

            ax_s = AX_SCALE * np.amax(np.abs(points[:, 3:6]))
        elif space is "SO3_GROUP":
            ax_s = AX_SCALE * np.amax(np.abs(points[:, :3]))
        elif space is "S2":
            ax_s = AX_SCALE
        else:
            raise NotImplementedError(

After Change

        sphere.draw(ax)

    elif space is "H2":
        poincare_disk = PoincareDisk()
        poincare_disk.add_points(points)
        poincare_disk.draw(points)

    return ax

Instances