diff --git a/sdmetrics/utils.py b/sdmetrics/utils.py index 30e56ebe..b50713ad 100644 --- a/sdmetrics/utils.py +++ b/sdmetrics/utils.py @@ -158,7 +158,7 @@ def fit(self, data): elif kind == 'O': # Categorical column. col_data = pd.DataFrame({'field': data[field]}) - enc = OneHotEncoder() + enc = OneHotEncoder(handle_unknown='ignore') enc.fit(col_data) self.column_transforms[field] = {'one_hot_encoder': enc} elif kind == 'M':