No progressbar on databricks

[henrytdsimmons]

compiled_jax = nutpie.compile_pymc_model(censored_model, backend="jax")
trace = nutpie.sample(
    compiled_jax,
    draws=1000,
    tune=500,
    chains=4,
)

No progress bar displays when running this command in a databricks notebook in .ipynb or .py formats.

pm.__version__='5.23.0'
nutpie.__version__='0.15.0'

[aseyboldt]

Thanks for reporting. I can't find any documentation about what API they have to show output, and without that, it's pretty hard to fix this.
Do you have any idea where I might find out more?

[henrytdsimmons]

The standard pymc one works, tqdm works. Sadly, I too have no idea where to find it out from databricks.