Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit a17c10b

Browse files
committed
case sensitivity on features
1 parent bda0936 commit a17c10b

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

splicemachine/features/feature_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1118,7 +1118,7 @@ def display_model_feature_drift(self, schema_name: str, table_name: str):
11181118
self._auth, params={ "schema": schema_name, "table": table_name})['metadata']
11191119

11201120
training_set_df, model_table_df = self._retrieve_model_data_sets(schema_name, table_name)
1121-
features = metadata['features'].split(',')
1121+
features = [f.upper() for f in metadata['features'].split(',')]
11221122
build_feature_drift_plot(features, training_set_df, model_table_df)
11231123

11241124

splicemachine/features/utils/drift_utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,8 @@ def build_feature_drift_plot(features, training_set_df, model_table_df):
7575
:param features: list of features to analyze
7676
:param training_set_df: the dataframe used for training the model that contains all the features to analyze
7777
:param model_table_df: the dataframe with the content of the model table containing all input features
78-
:return: None
7978
"""
80-
upper_features = [f.upper() for f in features]
81-
final_features = [f for f in upper_features if f in model_table_df.columns]
79+
final_features = [f for f in features if f in model_table_df.columns]
8280
# prep plot area
8381
n_bins = 15
8482
num_features = len(final_features)

0 commit comments

Comments
 (0)