first commit for mcse plot

src/arviz_plots/backend/matplotlib/__init__.py

@@ -5,6 +5,7 @@
 Sets ``zorder`` of all non-text "geoms" to ``2`` so that elements plotted later
 on are on top of previous ones.
 """
+
 import warnings
 from typing import Any, Dict
 
@@ -244,6 +245,42 @@ def scatter(
     return target.scatter(x, y, **_filter_kwargs(kwargs, None, artist_kws))
 
 
+def errorbar(
+    x,
+    y,
+    error,
+    target,
+    *,
+    size=unset,
+    marker=unset,
+    color=unset,
+    facecolor=unset,
+    edgecolor=unset,
+    width=unset,
+    **artist_kws,
+):
+    """Interface to matplotlib for an errorbar plot."""
+    artist_kws.setdefault("zorder", 2)
+    fillable_marker = (marker is unset) or (marker in Line2D.filled_markers)
+    if color is not unset:
+        if facecolor is unset and edgecolor is unset:
+            facecolor = color
+            if fillable_marker:
+                edgecolor = color
+        elif facecolor is unset:
+            facecolor = color
+        elif edgecolor is unset and fillable_marker:
+            edgecolor = color
+    kwargs = {
+        "capsize": size,
+        "marker": marker,
+        "markerfacecolor": facecolor,
+        "markeredgecolor": edgecolor,
+        "elinewidth": width,
+    }
+    return target.errorbar(x, y, error, **_filter_kwargs(kwargs, None, artist_kws))
+
+
 def text(
     x,
     y,

src/arviz_plots/backend/none/__init__.py

@@ -232,6 +232,49 @@ def scatter(
     return artist_element
 
 
+def errorbar(
+    x,
+    y,
+    error,
+    target,
+    *,
+    size=unset,
+    marker=unset,
+    color=unset,
+    facecolor=unset,
+    edgecolor=unset,
+    width=unset,
+    **artist_kws,
+):
+    """Interface to an errorbar plot."""
+    if color is not unset:
+        if facecolor is unset and edgecolor is unset:
+            facecolor = color
+            edgecolor = color
+        elif facecolor is unset:
+            facecolor = color
+        elif edgecolor is unset:
+            edgecolor = color
+    kwargs = {
+        "capsize": size,
+        "marker": marker,
+        "markerfacecolor": facecolor,
+        "markeredgecolor": edgecolor,
+        "elinewidth": width,
+    }
+    if not ALLOW_KWARGS and artist_kws:
+        raise ValueError("artist_kws not empty")
+    artist_element = {
+        "function": "errorbar",
+        "x": np.atleast_1d(x),
+        "y": np.atleast_1d(y),
+        "error": np.atleast_1d(error),
+        **_filter_kwargs(kwargs, artist_kws),
+    }
+    target.append(artist_element)
+    return artist_element
+
+
 def text(
     x,
     y,

src/arviz_plots/plots/__init__.py

@@ -2,8 +2,9 @@
 
 from .distplot import plot_dist
 from .forestplot import plot_forest
+from .mcseplot import plot_mcse
 from .ridgeplot import plot_ridge
 from .tracedistplot import plot_trace_dist
 from .traceplot import plot_trace
 
-__all__ = ["plot_dist", "plot_forest", "plot_trace", "plot_trace_dist", "plot_ridge"]
+__all__ = ["plot_dist", "plot_forest", "plot_trace", "plot_trace_dist", "plot_ridge", "plot_mcse"]