Protext against any return type

great-expectations · Sep 20, 2024 · 6719404 · 6719404
1 parent 2f2b458
commit 6719404
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/great_expectations/expectations/metrics/query_metrics/query_table.py b/great_expectations/expectations/metrics/query_metrics/query_table.py
@@ -77,7 +77,12 @@ def _sqlalchemy(
         result: Union[Sequence[sa.Row[Any]], Any] = execution_engine.execute_query(
             sa.select(sa.text(query))
         ).fetchmany(MAX_IN_MEMORY_RECORDS_ALLOWED)
-        return [element._asdict() for element in result]
+
+        if isinstance(result, (list, tuple)):
+            query_table_records = [element._asdict() for element in result]
+        else:
+            query_table_records = [result]
+        return query_table_records
         # </snippet>
 
     @metric_value(engine=SparkDFExecutionEngine)