From 7ecc615945b7bb48e103ca6eb278b39759d71c5a Mon Sep 17 00:00:00 2001 From: Shuchu Han Date: Sun, 8 Sep 2024 04:34:38 -0400 Subject: [PATCH] fix: Fix the mypy type check issue. (#4498) Signed-off-by: Shuchu Han --- .../infra/utils/postgres/connection_utils.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/sdk/python/feast/infra/utils/postgres/connection_utils.py b/sdk/python/feast/infra/utils/postgres/connection_utils.py index 3749fc2fc1..70148f3ee0 100644 --- a/sdk/python/feast/infra/utils/postgres/connection_utils.py +++ b/sdk/python/feast/infra/utils/postgres/connection_utils.py @@ -56,14 +56,14 @@ async def _get_connection_pool_async(config: PostgreSQLConfig) -> AsyncConnectio def _get_conninfo(config: PostgreSQLConfig) -> str: """Get the `conninfo` argument required for connection objects.""" - psycopg_config = { - "user": config.user, - "password": config.password, - "host": config.host, - "port": int(config.port), - "dbname": config.database, - } - return make_conninfo(conninfo="", **psycopg_config) + return make_conninfo( + conninfo="", + user=config.user, + password=config.password, + host=config.host, + port=int(config.port), + dbname=config.database, + ) def _get_conn_kwargs(config: PostgreSQLConfig) -> Dict[str, Any]: