

def get_train_vars(self, X):
    if (self.train_variables is not None):
        return X[self.train_variables]
    else:
        return X

dataset/ETHPy150Open anaderi/lhcb_trigger_ml/hep_ml/uboost.py uBoostClassifier.get_train_vars/original