def _spark_condition(cls, column, _metrics, threshold, double_sided, **kwargs): z_score, _, _ = _metrics["column_values.z_score.map"] if double_sided: threshold = abs(threshold) z_score = F.abs(z_score) return z_score < threshold
def _spark(cls, column_A, column_B, **kwargs): row_wise_cond = F.abs(column_A - column_B) == 3 return row_wise_cond