[WIP/TEST] prob_factory test running for the first time

Author: Leguark

gempy_probability/modules/model_definition/model_examples.py

@@ -80,7 +80,7 @@ def two_wells_prob_model_I(geo_model: gempy.core.data.GeoModel, normal, y_obs_li
 gravity_priors = {
         # e.g. density contrast of layer index 3
         # "mu_density": dist.Normal(2.62, 0.5),
-        "mu_thickness":  dist.Normal(
+        r'$\mu_{top}$':  dist.Normal(
             loc=torch.tensor(0.0487, dtype=torch.float64),
             scale=torch.tensor(0.1, dtype=torch.float64)
         )
@@ -89,7 +89,7 @@ def two_wells_prob_model_I(geo_model: gempy.core.data.GeoModel, normal, y_obs_li
 # 2) Wrap your forward pass into a small fn
 def run_gempy_forward(samples, geo_model):
 
-    mu_top = samples["mu_thickness"]
+    mu_top = samples["$\mu_{top}$"]
 
     interp_input = interpolation_input_from_structural_frame(geo_model)