e85cb30e6adb96462791fb07c084c87ac8f69c9f,pdpipe/col_generation.py,OneHotEncode,_fit_transform,#OneHotEncode#Any#Any#,242

Before Change

        return set(self._columns or []).issubset(df.columns)

    def _fit_transform(self, df, verbose):
        columns_to_encode = self._columns
        if self._columns is None:
            columns_to_encode = list(
                set(
                    df.select_dtypes(include=["object", "category"]).columns
                ).difference(self._exclude_columns)
            )
        if self._col_subset:
            columns_to_encode = [
                x for x in columns_to_encode if x in df.columns
            ]

After Change

        raise NotImplementedError

    def _fit_transform(self, df, verbose):
        columns_to_encode = self._get_columns(df, fit=True)
        self._cols_to_encode = columns_to_encode
        assign_map = {}
        if verbose:
            columns_to_encode = tqdm.tqdm(columns_to_encode)

Instances