def get_train_vars(self, X):
        if self.train_variables is None:
            return X
        else:
            return X[self.train_variables]
Gets the DataFrame and returns only columns
dataset/ETHPy150Open anaderi/lhcb_trigger_ml/hep_ml/uboost.py/uBoostBDT.get_train_vars