diff --git a/bambi/backend/inference_methods.py b/bambi/backend/inference_methods.py index 900d9c26..dee510e7 100644 --- a/bambi/backend/inference_methods.py +++ b/bambi/backend/inference_methods.py @@ -17,6 +17,9 @@ def __init__(self): self.pymc_methods = self._pymc_methods() def _get_bayeux_methods(self, model): + # If bayeux is not installed, return an empty MCMC list. + if model is None: + return {"mcmc": []} # Bambi only supports bayeux MCMC methods mcmc_methods = model.methods.get("mcmc") return {"mcmc": mcmc_methods} @@ -85,7 +88,7 @@ def bayeux_model(): A dummy model with a simple quadratic likelihood function. """ if importlib.util.find_spec("bayeux") is None: - return {"mcmc": []} + return None import bayeux as bx # pylint: disable=import-outside-toplevel