Skip to content

Commit ab4be33

Browse files
committed
[TEST] Run old test
1 parent f460733 commit ab4be33

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

gempy_probability/modules/model_definition/model_examples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def two_wells_prob_model_I(geo_model: gempy.core.data.GeoModel, normal, y_obs_li
4545

4646
# region Forward model computation
4747

48-
geo_model.counter +=1
48+
# geo_model.counter +=1
4949

5050
# * Compute the geological model
5151
geo_model.solutions = gempy_engine.compute_model(

tests/test_prob_model/test_prob_I.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,7 @@ def test_basic_gempy_I() -> None:
3838

3939
geo_model.grid.active_grids = gp.data.Grid.GridTypes.CUSTOM
4040
assert geo_model.grid.values.shape[0] == 100, "Custom grid should have 100 cells"
41-
geo_model.counter = 0
42-
gp.compute_model(gempy_model=geo_model)
41+
gp.compute_model(gempy_model=geo_model, validate_serialization=False)
4342
BackendTensor.change_backend_gempy(engine_backend=gp.data.AvailableBackends.PYTORCH)
4443

4544
normal = dist.Normal(
@@ -74,7 +73,7 @@ def _prob_run(geo_model: gp.data.GeoModel, prob_model: callable,
7473
num_samples=50
7574
)
7675
prior = predictive(geo_model, normal, y_obs_list)
77-
print("Number of interpolations: ", geo_model.counter)
76+
# print("Number of interpolations: ", geo_model.counter)
7877

7978
data = az.from_pyro(prior=prior)
8079
az.plot_trace(data.prior)
@@ -109,7 +108,7 @@ def _prob_run(geo_model: gp.data.GeoModel, prob_model: callable,
109108
data = az.from_pyro(posterior=mcmc, prior=prior, posterior_predictive=posterior_predictive)
110109
# endregion
111110

112-
print("Number of interpolations: ", geo_model.counter)
111+
# print("Number of interpolations: ", geo_model.counter)
113112

114113
if True: # * Save the arviz data
115114
data.to_netcdf("arviz_data.nc")

0 commit comments

Comments
 (0)