Simplify getting column name

Amsterdam · Feb 15, 2025 · ed91a0a · ed91a0a
1 parent 4ddba0f
commit ed91a0a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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)