99095d5412483ec623278bdb1a0c9e24b18bfc85,pdpipe/basic_stages.py,Bin,__init__,#Bin#,291

Before Change

    DEF_BIN_APP_MSG = "Binning columns {}..."

    def __init__(self, bin_map, drop=True, exraise=True):
        super(Bin, self).__init__(
            exraise=exraise,
            exmsg=Bin.DEF_BIN_EXC_MSG.format(
                list(bin_map.keys())),
            appmsg=Bin.DEF_BIN_APP_MSG.format(
                list(bin_map.keys()))
        )
        self._bin_map = bin_map
        self._drop = drop

    def _prec(self, df):

After Change

    def __init__(self, bin_map, drop=True, **kwargs):
        self._bin_map = bin_map
        self._drop = drop
        columns_str = _list_str(list(bin_map.keys()))
        super_kwargs = {
            "exmsg": Bin._DEF_BIN_EXC_MSG.format(columns_str),
            "appmsg": Bin._DEF_BIN_APP_MSG.format(
                "s" if len(bin_map) > 1 else "", columns_str),
            "desc": self._default_desc()
        }
        super_kwargs.update(**kwargs)
        super().__init__(**super_kwargs)

    def _prec(self, df):
        return set(self._bin_map.keys()).issubset(df.columns)

Instances