fix examples

arviz-devs · Feb 21, 2025 · 863d2d6 · 863d2d6
1 parent f85fcf7
commit 863d2d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/source/gallery/model_criticism/plot_ppc_rootogram.py b/docs/source/gallery/model_criticism/plot_ppc_rootogram.py
@@ -18,6 +18,8 @@
 dt = load_arviz_data("rugby")
 pc = azp.plot_ppc_rootogram(
     dt,
+    pc_kwargs={"aes": {"color": ["__variable__"]}}, # map variable to color
+    aes_map={"title": ["color"]}, # change title's color per variable
     backend="none",
 )
 pc.show()
diff --git a/src/arviz_plots/plots/ppcrootogramplot.py b/src/arviz_plots/plots/ppcrootogramplot.py
@@ -103,7 +103,7 @@ def plot_ppc_rootogram(
         >>> from arviz_plots import plot_ppc_rootogram, style
         >>> style.use("arviz-variat")
         >>> from arviz_base import load_arviz_data
-        >>> dt = load_arviz_data('crabs')
+        >>> dt = load_arviz_data('crabs_poisson')
         >>> plot_ppc_rootogram(dt)
 
 
@@ -206,8 +206,6 @@ def plot_ppc_rootogram(
 
     aes_map.setdefault("predictive_markers", plot_collection.aes_set)
     aes_map.setdefault("ci", plot_collection.aes_set)
-    # aes_map.setdefault("predictive_markers", ["color"])
-    # aes_map.setdefault("ci", ["color"])
     ## predictive_markers
     predictive_ms_kwargs = copy(plot_kwargs.get("predictive_markers", {}))