Updating docstring in _aggregate_shap_values_dict.

alteryx · Mar 3, 2021 · f8ffba7 · f8ffba7
1 parent 830775a
commit f8ffba7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/evalml/model_understanding/prediction_explanations/_algorithms.py b/evalml/model_understanding/prediction_explanations/_algorithms.py
@@ -131,8 +131,9 @@ def _aggreggate_shap_values_dict(values, provenance):
         values (dict):  A mapping of feature names to a list of SHAP values for each data point.
         provenance (dict): A mapping from a feature in the original data to the names of the features that were created
             from that feature.
+
     Returns:
-        dict
+        dict - mapping from feature name to shap values.
     """
 
     child_to_parent = {}