Add documentation notebook for slopes (#701)

bambi/plots/effects.py

@@ -1,12 +1,13 @@
+# pylint: disable=ungrouped-imports
 from dataclasses import dataclass, field
 import itertools
 from typing import Dict, Union
 
 import arviz as az
 import numpy as np
 import pandas as pd
-import xarray as xr
 from pandas.api.types import is_categorical_dtype, is_string_dtype
+import xarray as xr
 
 from bambi.models import Model
 from bambi.plots.create_data import create_cap_data, create_differences_data

bambi/plots/plot_types.py

@@ -46,7 +46,7 @@ def plot_numeric(
         ax = axes[0]
         values_main = transform_main(plot_data[main])
         ax.plot(values_main, y_hat_mean, solid_capstyle="butt", color="C0")
-        ax.fill_between(values_main, y_hat_bounds[0], y_hat_bounds[1], color="C0", alpha=0.4)
+        ax.fill_between(values_main, y_hat_bounds[0], y_hat_bounds[1], alpha=0.4)
     elif "group" in covariates and not "panel" in covariates:
         ax = axes[0]
         colors = get_unique_levels(plot_data[color])

bambi/plots/plotting.py

@@ -455,7 +455,6 @@ def plot_slopes(
 
     if conditional is None and average_by is None:
         raise ValueError("Must specify at least one of 'conditional' or 'average_by'.")
-
     if conditional is not None:
         if not isinstance(conditional, str):
             if len(conditional) > 3 and average_by is None:

bambi/plots/utils.py

@@ -356,7 +356,6 @@ def set_default_values(model: Model, data_dict: dict, kind: str) -> dict:
             if not isinstance(value, (list, np.ndarray)):
                 data_dict[key] = [value]
         return data_dict
-
     return data_dict
 
 

docs/examples.rst

@@ -27,4 +27,5 @@ Examples
   notebooks/hsgp_1d
   notebooks/hsgp_2d
   notebooks/plot_cap
-  notebooks/plot_comparisons
+  notebooks/plot_comparisons
+  notebooks/plot_slopes