def plt_equal(ax, limits=None):
    ax.set_box_aspect((1, 1, 1))
    set_axes_equal(ax, limits=limits)
def plt_equal(ax, limits=None):
    ax.set_box_aspect((1, 1, 1))
    set_axes_equal(ax, limits=limits)