Skip to content

Commit

Permalink
feature drift reduce output method fix (#2428)
Browse files Browse the repository at this point in the history
  • Loading branch information
yromanyshyn committed Mar 29, 2023
1 parent 73cb2d9 commit 7a4477d
Showing 1 changed file with 1 addition and 1 deletion.
Expand Up @@ -282,7 +282,7 @@ def run_logic(self, context: Context) -> CheckResult:
def reduce_output(self, check_result: CheckResult) -> Dict[str, float]:
"""Return an aggregated drift score based on aggregation method defined."""
feature_importance = pd.Series({column: info['Importance'] for column, info in check_result.value.items()})
feature_importance = None if None in feature_importance else feature_importance
feature_importance = None if None in feature_importance.values else feature_importance
values = pd.Series({column: info['Drift score'] for column, info in check_result.value.items()})
return self.feature_reduce(self.aggregation_method, values, feature_importance, 'Drift Score')

Expand Down

0 comments on commit 7a4477d

Please sign in to comment.