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