minor fixes

Author: aloctavodia

src/arviz_plots/plots/lm_plot.py

@@ -220,9 +220,7 @@ def plot_lm(
     observed_y = extract(dt, group="observed_data", var_names=y_obs, combined=False)
 
     if isinstance(ci_prob, (list | tuple | np.ndarray)):
-        x_with_prob = x_pred.expand_dims(dim={"prob": ci_prob})
-    else:
-        x_with_prob = x_pred
+        x_pred = x_pred.expand_dims(dim={"prob": ci_prob})
 
     plot_bknd = import_module(f".backend.{backend}", package="arviz_plots")
     bg_color = plot_bknd.get_background_color()
@@ -245,7 +243,7 @@ def plot_lm(
         pc_kwargs["aes"].setdefault("color", ["__variable__"])
         pc_kwargs = set_wrap_layout(pc_kwargs, plot_bknd, x_pred)
         plot_collection = PlotCollection.wrap(
-            x_with_prob,
+            x_pred,
             backend=backend,
             **pc_kwargs,
         )
@@ -420,7 +418,6 @@ def plot_lm(
     return plot_collection
 
 
-# This should be moved elsewhere
 def combine(x_pred, pe_value, ci_data, x_vars, y_vars, smooth, smooth_kwargs=None):
     """
     Combine  and sort x_pred, pe_value, ci_data into a dataset.
@@ -436,9 +433,10 @@ def combine(x_pred, pe_value, ci_data, x_vars, y_vars, smooth, smooth_kwargs=Non
 
     if smooth_kwargs is None:
         smooth_kwargs = {}
-        smooth_kwargs.setdefault("window_length", 55)
-        smooth_kwargs.setdefault("polyorder", 2)
-        smooth_kwargs.setdefault("n_points", 200)
+
+    smooth_kwargs.setdefault("window_length", 55)
+    smooth_kwargs.setdefault("polyorder", 2)
+    smooth_kwargs.setdefault("n_points", 200)
 
     for xv, yv in zip(x_vars, y_vars):
         old_dim = pe_value[yv].dims[0]