Update shap_plots.py

shap_plots.py

@@ -391,7 +391,7 @@ def shap_summary_plot(shap_values, features=None, feature_names=None, max_displa
     pl.gca().spines['top'].set_visible(False)
     pl.gca().spines['left'].set_visible(False)
     pl.gca().tick_params(color=axis_color, labelcolor=axis_color)
-    pl.yticks(range(len(feature_order)), labels=feature_names, fontsize=13)
+    pl.yticks(range(len(feature_order)), [feature_names[i] for i in feature_order], fontsize=13)
     pl.gca().tick_params('y', length=20, width=0.5, which='major')
     pl.gca().tick_params('x', labelsize=11)
     pl.ylim(-1, len(feature_order))