Skip to content

Commit 7a98a0e

Browse files
Fix to unify type comparison methods (#311)
* Use isinstance Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com> * Use isinstance for multiple types Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com> * Fix issues reported by flake8 Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com> Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>
1 parent 391bfc7 commit 7a98a0e

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

dice_ml/data_interfaces/base_data_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def _validate_and_set_outcome_name(self, params):
2020
if 'outcome_name' not in params:
2121
raise ValueError("should provide the name of outcome feature")
2222

23-
if type(params['outcome_name']) is str:
23+
if isinstance(params['outcome_name'], str):
2424
self.outcome_name = params['outcome_name']
2525
else:
2626
raise ValueError("should provide the name of outcome feature as a string")

dice_ml/data_interfaces/private_data_interface.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ def __init__(self, params):
3434
Default MAD value is 1 for all features.
3535
:param data_name (optional): Dataset name
3636
"""
37-
if sys.version_info > (3, 6, 0) and type(params['features']) in [dict, collections.OrderedDict]:
37+
if sys.version_info > (3, 6, 0) and isinstance(params['features'], (dict, collections.OrderedDict)):
3838
features_dict = params['features']
39-
elif sys.version_info <= (3, 6, 0) and type(params['features']) is collections.OrderedDict:
39+
elif sys.version_info <= (3, 6, 0) and isinstance(params['features'], collections.OrderedDict):
4040
features_dict = params['features']
4141
else:
4242
raise ValueError(
@@ -52,7 +52,7 @@ def __init__(self, params):
5252
self.categorical_levels = {}
5353

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

dice_ml/data_interfaces/public_data_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def _validate_and_set_continuous_features(self, params):
8080
if 'continuous_features' not in params:
8181
raise ValueError('continuous_features should be provided')
8282

83-
if type(params['continuous_features']) is list:
83+
if isinstance(params['continuous_features'], list):
8484
self.continuous_feature_names = params['continuous_features']
8585
else:
8686
raise ValueError(

0 commit comments

Comments
 (0)