revert change

pymc-devs · Dec 20, 2024 · d4e2bee · d4e2bee
1 parent c5a20fa
commit d4e2bee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc_bart/utils.py b/pymc_bart/utils.py
@@ -81,7 +81,7 @@ def plot_convergence(
     kind: str = "ecdf",
     figsize: Optional[tuple[float, float]] = None,
     ax=None,
-) -> plt.Axes:
+) -> list[plt.Axes]:
     """
     Plot convergence diagnostics.