Simplify sharing axes

SeldonIO · Sep 27, 2023 · 1786df7 · 1786df7
1 parent 5fb8dc8
commit 1786df7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alibi/explainers/partial_dependence.py b/alibi/explainers/partial_dependence.py
@@ -1097,8 +1097,8 @@ def _is_categorical(feature):
         min_val = min([ax_pd_lim[0] for _, ax_pd_lim in ax_group])
         max_val = max([ax_pd_lim[1] for _, ax_pd_lim in ax_group])
         axs = [ax[0] for ax in ax_group]
-        for ax1, ax2 in zip(axs, axs[1:]):
-            ax1.sharey(ax2)
+        for ax in axs[1:]:
+            ax.sharey(axs[0])
         axs[0].set_ylim(min_val, max_val)
 
     fig.set(**fig_kw)