I tried with Jupyter notebook and it's working fine.
And still not working ensure you use latest version of databricks-sql-connector
databricks-sql-connector
pip install --upgrade databricks-sql-connector