How to send max tree depth to bambi?

Hi, I have a Bambi model which hits the max treedepth. But when I try to put max_treedepth into the fit() command, it says I have given an unused step method.

Here is some sample code from Bambi 0.9.1:

import bambi
import pandas as pd
data =pd.DataFrame({"x": [1, 2, 3, 4, 5], "y": [9, 3, 4, 3, 4 ]})
model = bambi.Model("y~x", data)
results = = 12)

This returns “ValueError: Unused step method arguments: {‘max_treedepth’}”

Increasing max_treedepth is usually a pretty inefficient fix and most often does not fix anything. Did you try using more informative prior? It usually more effective to fix sampling difficulty.

Could you share the full traceback please? I think your code should work, but maybe there’s something I’m missing