From ed91a0a8bfc9c1148414bc89d470b1b5edf13e4a Mon Sep 17 00:00:00 2001 From: bas Date: Sat, 15 Feb 2025 21:32:56 +0100 Subject: [PATCH] Simplify getting column name --- src/dq_suite/validation.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/dq_suite/validation.py b/src/dq_suite/validation.py index e763f40..aa18027 100644 --- a/src/dq_suite/validation.py +++ b/src/dq_suite/validation.py @@ -140,12 +140,11 @@ def _get_gx_expectation_object(validation_rule: Rule, table_name: str): gx_expectation_class = getattr(gx_core, gx_expectation_name) gx_expectation_parameters: dict = validation_rule["parameters"] - if "column" not in gx_expectation_parameters.keys(): - gx_expectation_parameters["column"] = None + column_name = gx_expectation_parameters.get("column", None) gx_expectation_parameters["meta"] = { "table_name": table_name, - "column_name": gx_expectation_parameters['column'], + "column_name": column_name, "expectation_name": gx_expectation_name } return gx_expectation_class(**gx_expectation_parameters)