First commit for violinplot

arviz-devs · Sep 4, 2024 · 6fa6581 · 6fa6581
1 parent f4a39af
commit 6fa6581
Show file tree

Hide file tree

Showing 4 changed files with 455 additions and 2 deletions.
diff --git a/src/arviz_plots/backend/matplotlib/__init__.py b/src/arviz_plots/backend/matplotlib/__init__.py
@@ -372,7 +372,8 @@ def remove_axis(target, axis="y"):
         target.xaxis.set_ticks([])
         target.spines["left"].set_visible(True)
         target.spines["bottom"].set_visible(False)
-        target.xaxis.set_ticks_position("left")
+        # target.xaxis.set_ticks_position("left")
+        target.xaxis.set_ticks_position("bottom")
         target.tick_params(axis="y", direction="out", width=1, length=3)
     elif axis == "both":
         target.xaxis.set_ticks([])

diff --git a/src/arviz_plots/plots/__init__.py b/src/arviz_plots/plots/__init__.py
@@ -6,6 +6,7 @@
 from .ridgeplot import plot_ridge
 from .tracedistplot import plot_trace_dist
 from .traceplot import plot_trace
+from .violinplot import plot_violin
 
 __all__ = [
     "plot_compare",
@@ -14,4 +15,5 @@
     "plot_trace",
     "plot_trace_dist",
     "plot_ridge",
+    "plot_violin",
 ]