Formatting

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-12-24 17:26:09 -05:00
parent 686969eb86
commit e7a0a6ad85

View file

@ -757,8 +757,8 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM
new_cell['categories'][other_feature] = [0]
elif len(encoded) == 2:
new_cell['categories'][other_feature] = [1]
elif (other_feature not in new_cell['categories'].keys() or
len(new_cell['categories'][other_feature]) == 0):
elif (other_feature not in new_cell['categories'].keys()
or len(new_cell['categories'][other_feature]) == 0):
new_cell['categories'][other_feature] = [0, 1]
else:
if feature in cell['ranges'].keys():