"""Enum for data check action code."""
from enum import Enum
from evalml.utils import classproperty
[docs]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]