From 50be10212d18dcaeb54691bc41fc5a06ad4ed263 Mon Sep 17 00:00:00 2001 From: Maximilian Speicher Date: Sat, 3 Apr 2021 12:15:53 +0200 Subject: [PATCH] Add postgres as accepted connection kind --- awswrangler/postgresql.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/awswrangler/postgresql.py b/awswrangler/postgresql.py index e7eecf5a6..151eca61a 100644 --- a/awswrangler/postgresql.py +++ b/awswrangler/postgresql.py @@ -132,7 +132,7 @@ def connect( attrs: _db_utils.ConnectionAttributes = _db_utils.get_connection_attributes( connection=connection, secret_id=secret_id, catalog_id=catalog_id, dbname=dbname, boto3_session=boto3_session ) - if attrs.kind != "postgresql": + if attrs.kind != "postgresql" and attrs.kind != "postgres": raise exceptions.InvalidDatabaseType( f"Invalid connection type ({attrs.kind}. It must be a postgresql connection.)" )