diff --git a/integration_tests/tests/dbt_project.py b/integration_tests/tests/dbt_project.py index 67fa9d976..bb2f026b3 100644 --- a/integration_tests/tests/dbt_project.py +++ b/integration_tests/tests/dbt_project.py @@ -18,7 +18,7 @@ _DEFAULT_VARS = { "disable_dbt_invocation_autoupload": True, "disable_dbt_artifacts_autoupload": True, - "columns_upload_strategy": 'none', + "columns_upload_strategy": "none", "disable_run_results": True, "disable_freshness_results": True, "debug_logs": True, diff --git a/integration_tests/tests/test_dbt_artifacts/test_columns.py b/integration_tests/tests/test_dbt_artifacts/test_columns.py index 1a444147f..41238bef4 100644 --- a/integration_tests/tests/test_dbt_artifacts/test_columns.py +++ b/integration_tests/tests/test_dbt_artifacts/test_columns.py @@ -33,12 +33,12 @@ ) @Parametrization.case( name="only_with_description", - columns_upload_strategy='enriched_only', + columns_upload_strategy="enriched_only", expected_columns=["with_description"], ) @Parametrization.case( name="all", - columns_upload_strategy='all', + columns_upload_strategy="all", expected_columns=[ "with_description", "without_description", @@ -52,9 +52,7 @@ def test_flatten_table_columns( expected_columns: List[str], ) -> None: if columns_upload_strategy is not None: - dbt_project.dbt_runner.vars[ - "columns_upload_strategy" - ] = columns_upload_strategy + dbt_project.dbt_runner.vars["columns_upload_strategy"] = columns_upload_strategy flattened_columns = json.loads( dbt_project.dbt_runner.run_operation( "elementary.flatten_table_columns", macro_args={"table_node": TABLE_NODE}