enabling float for private data (#330)

Signed-off-by: Amit Sharma <[email protected]> Signed-off-by: Amit Sharma <[email protected]>
interpretml · Oct 10, 2022 · 7d8db2c · 7d8db2c
1 parent ed69a7a
commit 7d8db2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dice_ml/data_interfaces/private_data_interface.py b/dice_ml/data_interfaces/private_data_interface.py
@@ -52,7 +52,7 @@ def __init__(self, params):
         self.categorical_levels = {}
 
         for feature in features_dict:
-            if isinstance(features_dict[feature][0], int):  # continuous feature
+            if isinstance(features_dict[feature][0], (int, float)):  # continuous feature
                 self.continuous_feature_names.append(feature)
             else:
                 self.categorical_feature_names.append(feature)