yugabytedb smart driver support for Superset

[seethb]

Hi We tried using postgresql with yugabyteDB smart driver (pip install psycopg2-yugabytedb) to take care of connection load balancing. When we try with SQLAlchemy URI with below values

postgresql+psycopg2://yugabyte:[email protected]:5433/yugabyte?load_balance=true

We are getting below error when we add the DB connection settings in Superset UI.

ERROR: (builtins.NoneType) None [SQL: 'psycopg2.extensions.ConnectionInfo' object has no attribute 'host_addr'] (Background on this error at: https://sqlalche.me/e/14/dbapi)

It is supposed to accept the load_balance parameter to take care of load-balancing the connections? What else we are missing and needs to enable this driver for superset connection please.