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)
