utils
==================================

.. py:module:: evalml.data_checks.utils

.. autoapi-nested-parse::

   Utility methods for the data checks in EvalML.



Module Contents
---------------



Functions
~~~~~~~~~

.. autoapisummary::
   :nosignatures:

   evalml.data_checks.utils.handle_data_check_action_code



Contents
~~~~~~~~~~~~~~~~~~~
.. py:function:: handle_data_check_action_code(action_code)

   Handles data check action codes by either returning the DataCheckActionCode or converting from a str.

   :param action_code: Data check action code that needs to be handled.
   :type action_code: str or DataCheckActionCode

   :returns: DataCheckActionCode enum

   :raises KeyError: If input is not a valid DataCheckActionCode enum value.
   :raises ValueError: If input is not a string or DatCheckActionCode object.

   .. rubric:: Examples

   >>> assert handle_data_check_action_code("drop_col") == DataCheckActionCode.DROP_COL
   >>> assert handle_data_check_action_code("DROP_ROWS") == DataCheckActionCode.DROP_ROWS
   >>> assert handle_data_check_action_code("Impute_col") == DataCheckActionCode.IMPUTE_COL