format

Author: dlawin

data_diff/dbt.py

@@ -359,7 +359,7 @@ def _cloud_diff(diff_vars: TDiffVars, datasource_id: int, api: DatafoldAPI, org_
                 is_cloud=True,
                 org_id=org_meta.org_id,
                 org_name=org_meta.org_name,
-                user_id=org_meta.user_id
+                user_id=org_meta.user_id,
             )
             send_event_json(event_json)
 

tests/test_dbt.py

@@ -1,7 +1,6 @@
 import os
 
 from pathlib import Path
-import yaml
 from data_diff.cloud.datafold_api import TCloudApiDataSource
 from data_diff.cloud.datafold_api import TCloudApiOrgMeta
 from data_diff.diff_tables import Algorithm
@@ -615,7 +614,14 @@ def test_cloud_diff(self, mock_api, mock_os_environ, mock_print):
     @patch("data_diff.dbt.rich.print")
     @patch("data_diff.dbt.DatafoldAPI")
     def test_diff_is_cloud(
-        self, mock_api, mock_print, mock_dbt_parser, mock_cloud_diff, mock_local_diff, mock_get_diff_vars, mock_initialize_api,
+        self,
+        mock_api,
+        mock_print,
+        mock_dbt_parser,
+        mock_cloud_diff,
+        mock_local_diff,
+        mock_get_diff_vars,
+        mock_initialize_api,
     ):
         org_meta = TCloudApiOrgMeta(org_id=1, org_name="", user_id=1)
         connection = {}
@@ -832,7 +838,14 @@ def test_diff_only_prod_schema(
     @patch("data_diff.dbt.rich.print")
     @patch("data_diff.dbt.DatafoldAPI")
     def test_diff_is_cloud_no_pks(
-        self, mock_api, mock_print, mock_dbt_parser, mock_cloud_diff, mock_local_diff, mock_get_diff_vars, mock_initialize_api
+        self,
+        mock_api,
+        mock_print,
+        mock_dbt_parser,
+        mock_cloud_diff,
+        mock_local_diff,
+        mock_get_diff_vars,
+        mock_initialize_api,
     ):
         mock_dbt_parser_inst = Mock()
         mock_dbt_parser.return_value = mock_dbt_parser_inst