added WIP warning to pmx.fit

Author: Michal-Novomestsky

pymc_extras/inference/inla.py

@@ -1,3 +1,5 @@
+import warnings
+
 import arviz as az
 import numpy as np
 import pymc as pm
@@ -146,6 +148,8 @@ def fit_INLA(
     return_latent_posteriors: bool = True,
     **sampler_kwargs,
 ) -> az.InferenceData:
+    warnings.warn("Currently only valid for a nested normal model. WIP.", UserWarning)
+
     model = pm.modelcontext(model)
 
     # Check if latent field is Gaussian

tests/test_inla.py

@@ -26,6 +26,7 @@ def rng():
     return np.random.default_rng(seed)
 
 
+@pytest.mark.filterwarnings("ignore:Currently only valid for a nested normal model. WIP.")
 def test_non_gaussian_latent(rng):
     """
     INLA should raise an error if trying to use a non-Gaussian latent
@@ -45,6 +46,7 @@ def test_non_gaussian_latent(rng):
             pmx.fit(method="INLA", x=x)
 
 
+@pytest.mark.filterwarnings("ignore:Currently only valid for a nested normal model. WIP.")
 def test_non_precision_MvNormal(rng):
     """
     INLA should raise an error if trying to use a latent not in precision form
@@ -67,6 +69,7 @@ def test_non_precision_MvNormal(rng):
             pmx.fit(method="INLA", x=x)
 
 
+@pytest.mark.filterwarnings("ignore:Currently only valid for a nested normal model. WIP.")
 def test_3_layer_normal(rng):
     """
     Test INLA against a simple toy problem: