pre-commit fixes

Author: haritamar

integration_tests/docker-compose-dremio.yml

@@ -23,7 +23,7 @@ services:
     ports:
       - 9001:9001
       - 9000:9000
-    command: [ "server", "/data", "--console-address", ":9001" ]
+    command: ["server", "/data", "--console-address", ":9001"]
     volumes:
       - minio_data:/data
 
@@ -65,7 +65,7 @@ services:
       - dremio_data:/opt/dremio/data:rw
     # Workaround for permission issues in podman
     user: "0"
-  
+
   dremio-setup:
     image: alpine:latest
     container_name: dremio-setup
@@ -76,10 +76,10 @@ services:
     command: sh /dremio-setup.sh
     networks:
       - dremio-lakehouse
-      
+
 networks:
   dremio-lakehouse:
 
 volumes:
   dremio_data:
-  minio_data:
+  minio_data:

integration_tests/tests/dbt_project.py

@@ -7,9 +7,9 @@
 from uuid import uuid4
 
 from data_seeder import DbtDataSeeder
+from dbt_utils import get_database_and_schema_properties
 from elementary.clients.dbt.base_dbt_runner import BaseDbtRunner
 from elementary.clients.dbt.factory import create_dbt_runner
-from dbt_utils import get_database_and_schema_properties
 from logger import get_logger
 from ruamel.yaml import YAML
 
@@ -188,7 +188,9 @@ def test(
                 test_id, materialization
             )
         else:
-            database_property, schema_property = get_database_and_schema_properties(self.target)
+            database_property, schema_property = get_database_and_schema_properties(
+                self.target
+            )
             props_yaml = {
                 "version": 2,
                 "sources": [

integration_tests/tests/dbt_utils.py

@@ -3,4 +3,4 @@ def get_database_and_schema_properties(target: str, is_view: bool = False):
         return "datalake", "root_path"
     elif target == "clickhouse":
         return "schema", "schema"
-    return "database", "schema"
+    return "database", "schema"

integration_tests/tests/env.py

@@ -7,7 +7,7 @@ def __init__(self, target: str, project_dir: str):
         self.dbt_runner = dbt_project.get_dbt_runner(target, project_dir)
 
     def clear(self):
-        #drop schema in dremio doesnt work, but we run the dremio tests with docker so its not really important to drop the schema
+        # drop schema in dremio doesnt work, but we run the dremio tests with docker so its not really important to drop the schema
         if self.target != "dremio":
             self.dbt_runner.run_operation("elementary_tests.clear_env")
 

integration_tests/tests/test_dbt_artifacts/test_artifacts.py

@@ -96,8 +96,14 @@ def test_metrics_anomaly_score(dbt_project: DbtProject):
 
 @pytest.mark.requires_dbt_version("1.8.0")
 def test_source_freshness_results(test_id: str, dbt_project: DbtProject):
-    database_property, schema_property = get_database_and_schema_properties(dbt_project.target)
-    loaded_at_field = '"UPDATE_TIME"::timestamp' if dbt_project.target != "dremio" else "TO_TIMESTAMP(SUBSTRING(UPDATE_TIME, 0, 23), 'YYYY-MM-DD HH24:MI:SS.FFF')"
+    database_property, schema_property = get_database_and_schema_properties(
+        dbt_project.target
+    )
+    loaded_at_field = (
+        '"UPDATE_TIME"::timestamp'
+        if dbt_project.target != "dremio"
+        else "TO_TIMESTAMP(SUBSTRING(UPDATE_TIME, 0, 23), 'YYYY-MM-DD HH24:MI:SS.FFF')"
+    )
     source_config = {
         "version": 2,
         "sources": [