error : mesa.predict_proba

[shushan2017]

import time
from mesa import Mesa
from arguments import parser
from utils import Rater, load_dataset
from sklearn.tree import DecisionTreeClassifier

if name == 'main':

# load dataset & prepare environment
args = parser.parse_args()
rater = Rater(args.metric)
X_train, y_train, X_valid, y_valid, X_test, y_test = load_dataset(args.dataset)
base_estimator = DecisionTreeClassifier(max_depth=None)

# meta-training
print ('\nStart meta-training of MESA ... ...\n')
mesa = Mesa(
	args=args, 
	base_estimator=base_estimator, 
	n_estimators=args.max_estimators)
mesa.meta_fit(X_train, y_train, X_valid, y_valid, X_test, y_test)
mesa.predict_proba(X_test)

run....................

mesa.predict_proba(X_test)

File "D:\pyyj\mesa-master\environment.py", line 84, in predict_proba
if y_pred.shape[1] == 1:

IndexError: tuple index out of range