def _get_labels_right(self, validate=None): """Get all labels of the right dataframe.""" labels = [] for compare_func in self.features: labels = labels + listify(compare_func.labels_right) # check requested labels (for better error messages) if not is_label_dataframe(labels, validate): error_msg = "label is not found in the dataframe" raise KeyError(error_msg) return unique(labels)
def _get_labels(self, frame_i, validate=None): """Get all labels. Parameters ---------- frame_i : str A string, 'left' or 'right', incidating the dataframe to collect labels from. """ labels = [] for compare_func in self._compare_functions: labels = labels + listify(compare_func[frame_i]) # check requested labels (for better error messages) if not is_label_dataframe(labels, validate): error_msg = "label is not found in the dataframe" raise KeyError(error_msg) return unique(labels)