[BUG] Plot posterior function was a bit broken

Author: Leguark

examples/2-basic_geology/1-thickness_problem.py

@@ -1,4 +1,4 @@
-"""
+"""
 2.1 - Only Pyro
 ===============
 
@@ -148,7 +148,8 @@ def model(y_obs_list_):
 iteration = 99
 p.plot_posterior(
     prior_var=['$\\mu_{top}$', '$\\mu_{bottom}$'],
-    like_var=['$\mu_{thickness}$', '$\sigma_{thickness}$'],
+    like_var=['$\\mu_{top}$', '$\\mu_{bottom}$'],
+    # like_var=['$\\mu_{thickness}$', r"y_{top}"],
     obs='y_{thickness}',
     iteration=iteration,
     marginal_kwargs={

examples/DEP/ch5_2_introduction_pymc3.py

@@ -193,6 +193,7 @@ def plot_geo_setting_well():
 
 p.create_figure(figsize=(9, 5), joyplot=True, marginal=True, likelihood=True, n_samples=11)
 
-p.plot_posterior(['$\mu$', '$\sigma$'], ['$\mu$', '$\sigma$'], '$y$',
+p.plot_posterior(
+    ['$\mu$', '$\sigma$'], ['$\mu$', '$\sigma$'], '$y$',
                  marginal_kwargs={'plot_trace': False, 'credible_interval': .93, 'kind': 'kde'})
 plt.show()

gempy_probability/plot_posterior.py

@@ -521,12 +521,12 @@ def plot_joy(self, var_names: tuple = None, obs: Union[str, float] = None,
 
         obs = data.observed_data[obs] if type(obs) is str else obs
 
-        # data = convert_to_dataset(data, group="posterior")
         coords = {}
-        # var_names = _var_names(var_names, data)
-
         plotters_in_posterior = list(xarray_var_iter(
-            data=get_coords(data, coords),
+            data=get_coords(
+                convert_to_dataset(data, group="posterior"),
+                coords
+           ),
             var_names=var_names,
             combined=True
         ) )
@@ -538,8 +538,8 @@ def plot_joy(self, var_names: tuple = None, obs: Union[str, float] = None,
 
         plotters = plotters_in_posterior + plotters_in_posterior_predictive
 
-        x = plotters[0][3].flatten()
-        y = plotters[1][3].flatten()
+        x = plotters[0][-1].flatten()
+        y = plotters[1][-1].flatten()
 
         n_data = x.shape[0]
         # This is the special case if n_samples is smaller than the number of bells to plot