def METHOD_NAME(self, context: Context) -> CheckResult:
    """Calculate drift for all columns.
    Returns
    -------
    CheckResult
        value: drift score.
        display: label distribution graph, comparing the train and test distributions.
    """
    train_dataset = context.train.sample(self.n_samples, random_state=self.random_state)
    test_dataset = context.test.sample(self.n_samples, random_state=self.random_state)
    column_type = 'categorical' if context.task_type != TaskType.REGRESSION else 'numerical'
    return self._calculate_label_drift(train_dataset.label_col, test_dataset.label_col, train_dataset.label_name,
                                       column_type, context.with_display, (train_dataset.name, test_dataset.name))