Skip to content

Commit

Permalink
refactor long line
Browse files Browse the repository at this point in the history
  • Loading branch information
amesval committed Mar 24, 2024
1 parent e822388 commit f88f9f0
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/ml/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,9 @@ def performance_for_feature_slice(dataset, feature, model_attributes):
lb = model_attributes['lb']
cat_features = model_attributes['cat_features']
if feature not in cat_features:
raise ValueError("Invalid value. The feature should be a categorical feature.")
raise ValueError(
"Invalid value. The feature should be a categorical feature."
)
with open("./model/slice_output.txt", 'w') as f:
for value in dataset[feature].unique():
feat_filter = dataset[feature] == value
Expand Down

0 comments on commit f88f9f0

Please sign in to comment.