All tests pass, still need to review TODOs

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-05-29 19:15:16 +03:00
parent cc4cba0d8e
commit 26adcf3528

View file

@ -972,7 +972,8 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
if 'untouched' in cell:
for feature in cell['untouched']:
record_copy.pop(feature)
representative.pop(feature)
if feature in representative:
representative.pop(feature)
if record_copy == representative:
# handle numerical features
for feature in [key for key in cell['ranges'].keys() if