41d9e7e5442d0400d98c18b4ca56ae606f08f9e3,geomstats/visualization.py,,plot,#Any#Any#Any#,133

Before Change

                    "The plot function is not implemented"
                    " for the space {}.".format(space))

        ax = plt.subplot(111, projection="3d", aspect="equal")
        plt.setp(ax,
                 xlim=(-ax_s, ax_s),
                 ylim=(-ax_s, ax_s),
                 zlim=(-ax_s, ax_s),
                 xlabel="X", ylabel="Y", zlabel="Z")

    if space in ("SO3_GROUP", "SE3_GROUP"):
        trihedrons = convert_to_trihedron(points, space=space)
        for t in trihedrons:

After Change

                    "The plot function is not implemented"
                    " for the space {}.".format(space))

        if space is "H2":
            ax = plt.subplot(111, aspect="equal")
            plt.setp(ax,
                     xlim=(-ax_s, ax_s),
                     ylim=(-ax_s, ax_s),
                     xlabel="X", ylabel="Y")

        else:
            ax = plt.subplot(111, projection="3d", aspect="equal")
            plt.setp(ax,
                     xlim=(-ax_s, ax_s),
                     ylim=(-ax_s, ax_s),
                     zlim=(-ax_s, ax_s),
                     xlabel="X", ylabel="Y", zlabel="Z")

    if space in ("SO3_GROUP", "SE3_GROUP"):
        trihedrons = convert_to_trihedron(points, space=space)
        for t in trihedrons:
            t.draw(ax, **point_draw_kwargs)

Instances