add bounds to gallery images (#539)

* add bounds to gallery images * remove comment * adjust figsize * revert init_val skewt
arviz-devs · Sep 24, 2024 · bb7ef2b · bb7ef2b
1 parent d839f9d
commit bb7ef2b
Show file tree

Hide file tree

Showing 50 changed files with 77 additions and 3 deletions.
diff --git a/docs/examples/img/AsymmetricLaplace.png b/docs/examples/img/AsymmetricLaplace.png
diff --git a/docs/examples/img/Bernoulli.png b/docs/examples/img/Bernoulli.png
diff --git a/docs/examples/img/Beta.png b/docs/examples/img/Beta.png
diff --git a/docs/examples/img/BetaBinomial.png b/docs/examples/img/BetaBinomial.png
diff --git a/docs/examples/img/BetaScaled.png b/docs/examples/img/BetaScaled.png
diff --git a/docs/examples/img/Binomial.png b/docs/examples/img/Binomial.png
diff --git a/docs/examples/img/Categorical.png b/docs/examples/img/Categorical.png
diff --git a/docs/examples/img/Cauchy.png b/docs/examples/img/Cauchy.png
diff --git a/docs/examples/img/Censored.png b/docs/examples/img/Censored.png
diff --git a/docs/examples/img/ChiSquared.png b/docs/examples/img/ChiSquared.png
diff --git a/docs/examples/img/DiscreteUniform.png b/docs/examples/img/DiscreteUniform.png
diff --git a/docs/examples/img/DiscreteWeibull.png b/docs/examples/img/DiscreteWeibull.png
diff --git a/docs/examples/img/ExGaussian.png b/docs/examples/img/ExGaussian.png
diff --git a/docs/examples/img/Exponential.png b/docs/examples/img/Exponential.png
diff --git a/docs/examples/img/Gamma.png b/docs/examples/img/Gamma.png
diff --git a/docs/examples/img/Geometric.png b/docs/examples/img/Geometric.png
diff --git a/docs/examples/img/Gumbel.png b/docs/examples/img/Gumbel.png
diff --git a/docs/examples/img/HalfCauchy.png b/docs/examples/img/HalfCauchy.png
diff --git a/docs/examples/img/HalfNormal.png b/docs/examples/img/HalfNormal.png
diff --git a/docs/examples/img/HalfStudentT.png b/docs/examples/img/HalfStudentT.png
diff --git a/docs/examples/img/Hurdle.png b/docs/examples/img/Hurdle.png
diff --git a/docs/examples/img/HyperGeometric.png b/docs/examples/img/HyperGeometric.png
diff --git a/docs/examples/img/InverseGamma.png b/docs/examples/img/InverseGamma.png
diff --git a/docs/examples/img/Kumaraswamy.png b/docs/examples/img/Kumaraswamy.png
diff --git a/docs/examples/img/Laplace.png b/docs/examples/img/Laplace.png
diff --git a/docs/examples/img/LogLogistic.png b/docs/examples/img/LogLogistic.png
diff --git a/docs/examples/img/LogNormal.png b/docs/examples/img/LogNormal.png
diff --git a/docs/examples/img/Logistic.png b/docs/examples/img/Logistic.png
diff --git a/docs/examples/img/LogitNormal.png b/docs/examples/img/LogitNormal.png
diff --git a/docs/examples/img/Moyal.png b/docs/examples/img/Moyal.png
diff --git a/docs/examples/img/NegativeBinomial.png b/docs/examples/img/NegativeBinomial.png
diff --git a/docs/examples/img/Normal.png b/docs/examples/img/Normal.png
diff --git a/docs/examples/img/Pareto.png b/docs/examples/img/Pareto.png
diff --git a/docs/examples/img/Poisson.png b/docs/examples/img/Poisson.png
diff --git a/docs/examples/img/Rice.png b/docs/examples/img/Rice.png
diff --git a/docs/examples/img/SkewNormal.png b/docs/examples/img/SkewNormal.png
diff --git a/docs/examples/img/SkewStudentT.png b/docs/examples/img/SkewStudentT.png
diff --git a/docs/examples/img/StudentT.png b/docs/examples/img/StudentT.png
diff --git a/docs/examples/img/Triangular.png b/docs/examples/img/Triangular.png
diff --git a/docs/examples/img/Truncated.png b/docs/examples/img/Truncated.png
diff --git a/docs/examples/img/TruncatedNormal.png b/docs/examples/img/TruncatedNormal.png
diff --git a/docs/examples/img/Uniform.png b/docs/examples/img/Uniform.png
diff --git a/docs/examples/img/VonMises.png b/docs/examples/img/VonMises.png
diff --git a/docs/examples/img/Wald.png b/docs/examples/img/Wald.png
diff --git a/docs/examples/img/Weibull.png b/docs/examples/img/Weibull.png
diff --git a/docs/examples/img/ZeroInflatedBinomial.png b/docs/examples/img/ZeroInflatedBinomial.png
diff --git a/docs/examples/img/ZeroInflatedNegativeBinomial.png b/docs/examples/img/ZeroInflatedNegativeBinomial.png
diff --git a/docs/examples/img/ZeroInflatedPoisson.png b/docs/examples/img/ZeroInflatedPoisson.png
diff --git a/docs/get_cover_gallery.py b/docs/get_cover_gallery.py
@@ -12,9 +12,10 @@
 rng = np.random.default_rng(247)
 
 init_vals["Hurdle"] = None
+init_vals["SkewStudentT"] = {"mu": 0.0, "sigma": 1, "a": 2.5, "b": 1.5}
 for name, params in init_vals.items():
     color = f"C{rng.integers(0, 4)}"
-    _, ax = plt.subplots(figsize=(6, 4))
+    _, ax = plt.subplots(figsize=(3.5, 2.3))
     dist = getattr(distributions, name)
     if name in ["Truncated", "Censored"]:
         dist(Gamma(mu=2, sigma=1), -np.inf, 2).plot_pdf(legend=False, ax=ax, color=color)
@@ -42,5 +43,78 @@
 
     ax.set_xticks([])
     ax.set_yticks([])
-    ax.spines[["left", "bottom"]].set_visible(False)
+    ax.spines[["left"]].set_visible(False)
+
+    if name in ["Categorical"]:
+        ax.set_xticks([0, 1, 2], labels=["♣", "♥", "♦"])
+    else:
+        if name in ["BetaScaled", "Truncated", "Censored", "Pareto"]:
+            l_b, u_b = (-np.inf, np.inf)
+        elif name == "Hurdle":
+            l_b, u_b = (0, np.inf)
+        else:
+            l_b, u_b = dist().support
+
+        if l_b == -np.inf:
+            l_b = None
+        if u_b == np.inf:
+            u_b = None
+
+        pos = ax.get_ylim()[0]
+
+        # The boundaries depended on the parameterization
+        if name in [
+            "BetaScaled",
+            "TruncatedNormal",
+            "Triangular",
+            "Uniform",
+            "Binomial",
+            "BetaBinomial",
+            "DiscreteUniform",
+            "HyperGeometric",
+            "Censored",
+            "Hurdle",
+            "Truncated",
+        ]:
+            marker_0 = marker_1 = "*"
+        elif name == "Pareto":
+            marker_0 = "*"
+            marker_1 = ">"
+        else:
+            marker_0 = "<"
+            marker_1 = ">"
+
+        if l_b is None:
+            ax.plot(
+                0,
+                pos,
+                color="k",
+                marker=marker_0,
+                transform=ax.get_yaxis_transform(),
+                clip_on=False,
+                zorder=3,
+            )
+        if u_b is None:
+            ax.plot(
+                1,
+                pos,
+                color="k",
+                marker=marker_1,
+                transform=ax.get_yaxis_transform(),
+                clip_on=False,
+                zorder=3,
+            )
+
+        if name == "VonMises":
+            ax.set_xticks([l_b, u_b], labels=["$-\pi$", "$\pi$"])
+
+        if l_b is not None and u_b is not None:
+            ax.set_xticks([l_b, u_b])
+        elif l_b is not None and u_b is None:
+            ax.set_xticks([l_b])
+        elif l_b is None and u_b is not None:
+            ax.set_xticks([u_b])
+
+        ax.spines["bottom"].set_position(("data", pos))
+
     plt.savefig(f"examples/img/{name}.png")
diff --git a/preliz/internal/distribution_helper.py b/preliz/internal/distribution_helper.py
@@ -97,7 +97,7 @@ def process_extra(input_string):
     "Pareto": {"alpha": 5, "m": 2.0},
     "Rice": {"nu": 2.0, "sigma": 1.0},
     "SkewNormal": {"mu": 0.0, "sigma": 1, "alpha": 6.0},
-    "SkewStudentT": {"mu": 0.0, "sigma": 1, "a": 2.0, "b": 2.0},
+    "SkewStudentT": {"mu": 0.0, "sigma": 1, "a": 2, "b": 2},
     "StudentT": {"nu": 7, "mu": 0.0, "sigma": 1},
     "Triangular": {"lower": -2, "c": 0.0, "upper": 2.0},
     "TruncatedNormal": {"mu": 0.0, "sigma": 1, "lower": -2, "upper": 3.0},