def METHOD_NAME(cls, ctx, op: "StatsModelsTrain"):
    if op.merge:  # pragma: no cover
        super().METHOD_NAME(ctx, op)
    elif op.stage == OperandStage.combine:
        cls._execute_combine(ctx, op)
    else:
        cls._execute_map(ctx, op)