Сломанные оси AttributeError: объект «SubplotSpec» не имеет атрибута «is_last_row»

Я пытаюсь использовать сломанные оси python, чтобы сломать оси x и y. Я запускаю код, предложенный здесь https://www.codespeedy.com/create-a-plot-with-broken-axis-in-python-using-matplotlib/

import matplotlib.pyplot as plt 
from brokenaxes import brokenaxes 
import numpy as np 
fig = plt.figure(figsize=(6,4)) 
baxes = brokenaxes(xlims=((-2,3),(5,8)), ylims=((0,8),(9.5,21)), hspace=.1) 
X = np.array([3,-1,0,4,5,-2,7]) 
Y = x**2
Z = x**3
baxes.plot(X,Y,label = "squared") 
baxes.plot(X,Z,label = "cubed") 
baxes.legend(loc = "best")
plt.plot()
plt.show()

Однако я получаю сообщение об ошибке:

AttributeError: 'SubplotSpec' object has no attribute 'is_last_row'