Skip to content

Commit 7d8db2c

Browse files
authored
enabling float for private data (#330)
Signed-off-by: Amit Sharma <amit_sharma@live.com> Signed-off-by: Amit Sharma <amit_sharma@live.com>
1 parent ed69a7a commit 7d8db2c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

dice_ml/data_interfaces/private_data_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def __init__(self, params):
5252
self.categorical_levels = {}
5353

5454
for feature in features_dict:
55-
if isinstance(features_dict[feature][0], int): # continuous feature
55+
if isinstance(features_dict[feature][0], (int, float)): # continuous feature
5656
self.continuous_feature_names.append(feature)
5757
else:
5858
self.categorical_feature_names.append(feature)

0 commit comments

Comments
 (0)