python报错“too many indices for array”

在复现一篇文章，用一个新算法“Survival Quilts ”来预测前列腺癌死亡率
在网上找到了他共享的数据和代码，https://github.com/chl8856/SurvivalQuilts
但自己跑下来报“too many indices for array: array is 1-dimensional, but 2 were indexed”
有人能帮忙看下吗？

initial training of underlying models...
CV.. 1/10
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_15028/761714535.py in <module>
      1 model_sq = SurvivalQuilts()
----> 2 model_sq.train(tr_X, tr_T, tr_Y)

D:\python file\SurvivalQuilts-master\class_SurvivalQuilts.py in train(self, X, T, Y)
     55         for cv_idx in range(self.num_cv):
     56             print('CV.. {}/{}'.format(cv_idx+1, self.num_cv))
---> 57             pulled_models, tmp_CINDEX, tmp_BRIER = self._get_models_pulled_CV(X, T, Y, seed=cv_idx)
     58 
     59             metric_CINDEX[cv_idx,:,:] = tmp_CINDEX

D:\python file\SurvivalQuilts-master\class_SurvivalQuilts.py in _get_models_pulled_CV(self, X, T, Y, seed)
    287 
    288         for m, model in enumerate(pulled_models):
--> 289             pred = model.predict(X_va, self.time_horizons)
    290 
    291             for t, eval_time in enumerate(self.time_horizons):

D:\python file\SurvivalQuilts-master\class_UnderlyingModels.py in predict(self, X, time_horizons)
     51                     preds_[:, t] = 1. - surv[:, 0]
     52                 else:
---> 53                     preds_[:, t] = 1. - surv[:, tmp_time[0]]
     54 
     55         else:

IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

一维的向量, 你用了两层索引