258bb52cbbf7ebcf0d0e6d2779dd9a86ea323e7f,janitor/functions.py,,encode_categorical,#Any#Any#,334

Before Change

    
    if isinstance(column_names, list) or isinstance(column_names, tuple):
        for col in column_names:
            assert col in df.columns, JanitorError(
                "{col} missing from dataframe columns!".format(col=col)
            )
            df[col] = pd.Categorical(df[col])
    elif isinstance(column_names, str):
        assert column_names in df.columns, JanitorError(
            "{column_names} missing from dataframe columns!".format(

After Change

    
    if isinstance(column_names, list) or isinstance(column_names, tuple):
        for col in column_names:
            if col not in df.columns:
                raise JanitorError(f"{col} missing from dataframe columns!")
            df[col] = pd.Categorical(df[col])
    elif isinstance(column_names, str):
        if column_names not in df.columns:
            raise JanitorError(

Instances