run.py

from typing import reveal_type
import bambi as bmb
import pandas as pd

df = pd.read_csv("mtcars.csv")
model = bmb.Model("mpg ~ wt + hp + disp", df)

trace = model.fit()
reveal_type(trace)

trace_vi = model.fit(inference_method="vi")
reveal_type(trace_vi)

trace_laplace = model.fit(inference_method="laplace")
reveal_type(trace_laplace)

foo = model.predict(trace, inplace=False)
reveal_type(foo)

trace_prior = model.prior_predictive()
reveal_type(trace_prior)