Update typechecking for matplotlib 3.8.1

SeldonIO · Nov 2, 2023 · b869eff · b869eff
1 parent 71cc77a
commit b869eff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/alibi/explainers/tests/test_partial_dependence.py b/alibi/explainers/tests/test_partial_dependence.py
@@ -637,7 +637,7 @@ def assert_pd_ice_values(feature: int, target_idx: int, kind: str, explanation:
         line = ax.lines[0] if kind == 'average' else ax.lines[2]
         assert_pd_values(feature_values=explanation.data['feature_values'][feature],
                          pd_values=explanation.data['pd_values'][feature][target_idx],
-                         line=line)  # type: ignore[arg-type]
+                         line=line)
 
     if kind in ['individual', 'both']:
         # check the ice values

diff --git a/alibi/utils/visualization.py b/alibi/utils/visualization.py
@@ -354,7 +354,7 @@ def _create_heatmap(data: np.ndarray,
     plt.setp(ax.get_xticklabels(), rotation=90, ha="right", rotation_mode="anchor")
 
     # turn spines off and create white grid.
-    ax.spines[:].set_visible(False)  # type: ignore[call-overload]
+    ax.spines[:].set_visible(False)
 
     ax.set_xticks(np.arange(data.shape[1]+1)-.5, minor=True)
     ax.set_yticks(np.arange(data.shape[0]+1)-.5, minor=True)