Preventing tabulate from truncating in Google Colab?

[edmuthiah]

Hello, is there a way to prevent Rich Table output from the .tabulate method from being truncated in Google Colab? For example I have tried setting the width like so:

output = model.tabulate(
    jax.random.key(0), 
    batch, 
    show_repeated=True, 
    table_kwargs={"width": 120} 
)  
print(output)

But it results in:

[edmuthiah]

Solved by setting console_kwargs=dict(width=10_000)))