pre-commit

elementary-data · Jan 7, 2025 · 07209da · 07209da
1 parent b2ba1d6
commit 07209da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git 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
@@ -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}