<gh_stars>0
"""Enum for data check action code."""
from enum import Enum

from evalml.utils import classproperty


class DataCheckActionCode(Enum):
    """Enum for data check action code."""

    DROP_COL = "drop_col"
    """Action code for dropping a column."""

    DROP_ROWS = "drop_rows"
    """Action code for dropping rows."""

    IMPUTE_COL = "impute_col"
    """Action code for imputing a column."""

    TRANSFORM_TARGET = "transform_target"
    """Action code for transforming the target data."""

    @classproperty
    def _all_values(cls):
        return {code.value.upper(): code for code in list(cls)}

    def __str__(self):
        """String representation of the DataCheckActionCode enum."""
        datacheck_action_code_dict = {
            DataCheckActionCode.DROP_COL.name: "drop_col",
            DataCheckActionCode.DROP_ROWS.name: "drop_rows",
            DataCheckActionCode.IMPUTE_COL.name: "impute_col",
            DataCheckActionCode.TRANSFORM_TARGET.name: "transform_target",
        }
        return datacheck_action_code_dict[self.name]

"""Enum for data check action code."""
from enum import Enum

from evalml.utils import classproperty


class DataCheckActionCode(Enum):
    """Enum for data check action code."""

    DROP_COL = "drop_col"
    """Action code for dropping a column."""

    DROP_ROWS = "drop_rows"
    """Action code for dropping rows."""

    IMPUTE_COL = "impute_col"
    """Action code for imputing a column."""

    TRANSFORM_TARGET = "transform_target"
    """Action code for transforming the target data."""

    @classproperty
    def _all_values(cls):
        return {code.value.upper(): code for code in list(cls)}

    def __str__(self):
        """String representation of the DataCheckActionCode enum."""
        datacheck_action_code_dict = {
            DataCheckActionCode.DROP_COL.name: "drop_col",
            DataCheckActionCode.DROP_ROWS.name: "drop_rows",
            DataCheckActionCode.IMPUTE_COL.name: "impute_col",
            DataCheckActionCode.TRANSFORM_TARGET.name: "transform_target",
        }
        return datacheck_action_code_dict[self.name]
Enum for data check action code. Enum for data check action code. Action code for dropping a column. Action code for dropping rows. Action code for imputing a column. Action code for transforming the target data. String representation of the DataCheckActionCode enum.