Currently databricks output requires you to call .arrow() which does a collect
Currently databricks output requires you to call .arrow() which does a collect