Pyright fixes for plot.py

sbi-dev · Dec 4, 2023 · d25a46b · d25a46b
1 parent 8d1f0d0
commit d25a46b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sbi/analysis/plot.py b/sbi/analysis/plot.py
@@ -954,7 +954,7 @@ def _get_default_opts():
         "samples_colors": plt.rcParams["axes.prop_cycle"].by_key()["color"][0::2],
         "points_colors": plt.rcParams["axes.prop_cycle"].by_key()["color"][1::2],
         # ticks
-        "tickformatter": mpl.ticker.FormatStrFormatter("%g"),
+        "tickformatter": mpl.ticker.FormatStrFormatter("%g"),  # type: ignore
         "tick_labels": None,
         # options for hist
         "hist_diag": {
@@ -1068,7 +1068,7 @@ def _sbc_rank_plot(
     show_ylabel: bool = False,
     sharey: bool = False,
     fig: Optional[Figure] = None,
-    ax: Optional[Axes] = None,
+    ax=None,  # no type hint to avoid hassle with pyright. Should be `array(Axes).`
     figsize: Optional[tuple] = None,
 ) -> Tuple[Figure, Axes]:
     """Plot simulation-based calibration ranks as empirical CDFs or histograms.