46aea54b80a6f7516447a84b3d66d9502fe3834a,pdpipe/basic_stages.py,ValKeep,__init__,#ValKeep#Any#Any#,141

Before Change

    def __init__(self, values, columns=None, **kwargs):
        self._values = values
        self._values_str = _list_str(self._values)
        self._columns_str = _list_str(columns)
        if columns is None:
            self._columns = None
            apply_msg = ValKeep._DEF_VALKEEP_APPLY_MSG.format(
                self._values_str)
        else:
            self._columns = _interpret_columns_param(columns)
            apply_msg = ValKeep._DEF_VALKEEP_APPLY_MSG.format(
                "{} in {}".format(
                    self._values_str, self._columns_str))
        super_kwargs = {
            "exmsg": ValKeep._DEF_VALKEEP_EXC_MSG.format(self._columns_str),
            "appmsg": apply_msg,
            "desc": self._default_desc()
        }
        super_kwargs.update(**kwargs)
        super().__init__(**super_kwargs)

    def _prec(self, df):

After Change

    def __init__(self, values, columns=None, **kwargs):
        self._values = values
        self._values_str = _list_str(self._values)
        super_kwargs = {
            "columns": columns,
            "desc_temp": "Keep values {} in columns {{}}".format(
                self._values_str),
        }
        super_kwargs.update(**kwargs)
        super_kwargs["none_columns"] = "all"
        super().__init__(**super_kwargs)

    def _transformation(self, df, verbose, fit):
        inter_df = df

Instances