From 6bbda626b542a6432fb60845fadefc482a1e2b95 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Juan=20Carlos=20Jos=C3=A9=20Camacho?= Date: Wed, 8 Nov 2023 13:18:16 -0600 Subject: [PATCH] [DH-4966] Fix scanning table list index out of range error for postgres (#242) --- dataherald/db_scanner/sqlalchemy.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/dataherald/db_scanner/sqlalchemy.py b/dataherald/db_scanner/sqlalchemy.py index 26dff4b9..dd29195d 100644 --- a/dataherald/db_scanner/sqlalchemy.py +++ b/dataherald/db_scanner/sqlalchemy.py @@ -97,7 +97,10 @@ def get_processed_column( # noqa: PLR0911 rs = db_engine.engine.execute( f"SELECT n_distinct, most_common_vals::TEXT::TEXT[] FROM pg_catalog.pg_stats WHERE tablename = '{table}' AND attname = '{column['name']}'" # noqa: S608 E501 ).fetchall() - if MIN_CATEGORY_VALUE < rs[0]["n_distinct"] <= MAX_CATEGORY_VALUE: + if ( + len(rs) > 0 + and MIN_CATEGORY_VALUE < rs[0]["n_distinct"] <= MAX_CATEGORY_VALUE + ): return ColumnDetail( name=column["name"], data_type=str(column["type"]),