list index out of range for regressor with y in {0,1} about dtreeviz HOT 8 CLOSED

--------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-40-4568def0d1f4> in <module>
     76                       azim=35,
     77                       dist=12,
---> 78                       show={'splits','title'})
     79 
     80 

~/anaconda3/lib/python3.7/site-packages/dtreeviz/trees.py in rtreeviz_bivar_3D(ax, X_train, y_train, max_depth, feature_names, target_name, fontsize, ticks_fontsize, fontname, azim, elev, dist, show, colors, n_colors_in_map)
    246 
    247     for node, bbox in tesselation:
--> 248         plane(node, bbox)
    249 
    250     x, y, z = X_train[:, 0], X_train[:, 1], y_train

~/anaconda3/lib/python3.7/site-packages/dtreeviz/trees.py in plane(node, bbox)
    230         # print(f"{color_map[int(((node.prediction()-y_lim[0])/y_range)*(n_colors_in_map-1))]}")
    231         ax.plot_surface(xx, yy, z, alpha=.85, shade=False,
--> 232                         color=color_map[int(((node.prediction()-y_lim[0])/y_range)*(n_colors_in_map-1))],
    233                         edgecolor=colors['edge'], lw=.3)
    234 

IndexError: list index out of range

from sklearn.datasets import *
from sklearn import tree
from dtreeviz.trees import *

from mpl_toolkits.mplot3d import Axes3D




data = pd.DataFrame({'Edad': [ 17, 64, 18, 20, 38, 49, 55, 25, 29,  31, 33, 52, 65, 27, 39,  54, 30, 28, 27, 18, 47],
                     'Sexo': ['H', 'M', 'M', 'M', 'M', 'H', 'H', 'M', 'H', 'H', 'H', 'M', 'M', 'M', 'M',  'H', 'H', 'H', 'M', 'H', 'M'],
                      'Salario': [25, 80, 22, 36, 37, 59, 74, 70, 33, 102, 88, 35, 70, 51, 38,  94, 50, 35, 74, 32, 45], 
                      'Pagador': [1,   0,  1,  0,  1,  0,  0,  1,  1,   0,  1,  0,  0,  1,  1,   0,  1,  0,  0,  1,  0]})


from sklearn import preprocessing
code = preprocessing.LabelEncoder()
code.fit(["H", "M"])
data['Sexo']= code.transform(data['Sexo'])
data = data.sort_values('Edad')


features = [2,0]
X = data.values[:,features]
print(X)


y = data['Pagador'].values
print(y)

figsize = (42,16)
fig = plt.figure(figsize=figsize)
ax = fig.add_subplot(111, projection='3d')
t = rtreeviz_bivar_3D(ax,
                      X,y,
                      max_depth=None,
                      feature_names=['Edad','Sexo'],
                      target_name='Pagador',
                      fontsize=11,
                      elev=30,
                      azim=35,
                      dist=12,
                      show={'splits','title'})



plt.show()

ctreeviz_bivar(ax, X, y, max_depth=5,
               feature_names=['Edad', 'Sexo'],
               class_names=['yes', 'no'],
               target_name='Pagador',
               show={'splits', 'title'},
               colors={'scatter_edge': 'black'}
               )