Skip to content

Commit

Permalink
Fix bug when resetting TreeShap predictor
Browse files Browse the repository at this point in the history
  • Loading branch information
jklaise committed Nov 17, 2021
1 parent 6429778 commit fbab8e7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion alibi/explainers/shap_wrappers.py
Original file line number Diff line number Diff line change
Expand Up @@ -1594,4 +1594,4 @@ def _check_result_summarisation(self,

def reset_predictor(self, predictor: Any) -> None:
# TODO: check what else should be done (e.g. validate dtypes again?)
self.model = predictor
self.predictor = predictor

0 comments on commit fbab8e7

Please sign in to comment.