99095d5412483ec623278bdb1a0c9e24b18bfc85,pdpipe/basic_stages.py,Encode,__init__,#Encode#,554

Before Change

                exclude_columns, "exclude_columns")
        self._drop = drop
        self.encoders = {}
        super(Encode, self).__init__(
            exraise=exraise,
            exmsg=Encode.DEF_ENCODE_EXC_MSG.format(self._columns),
            appmsg=Encode.DEF_ENCODE_APP_MSG.format(self._columns)
        )

    def _prec(self, df):
        return set(self._columns or []).issubset(df.columns)

After Change

                exclude_columns, "exclude_columns")
        self._drop = drop
        self.encoders = {}
        col_str = _list_str(self._columns)
        super_kwargs = {
            "exmsg": Encode._DEF_ENCODE_EXC_MSG.format(col_str),
            "appmsg": Encode._DEF_ENCODE_APP_MSG.format(col_str),
            "desc": "Encode {}".format(col_str or "all categorical columns")
        }
        super_kwargs.update(**kwargs)
        super().__init__(**super_kwargs)

    def _prec(self, df):
        return set(self._columns or []).issubset(df.columns)

Instances