Ran black

Author: erezsh

data_diff/__init__.py

@@ -73,15 +73,19 @@ def diff_tables(
 
     """
     tables = [table1, table2]
-    override_attrs = {k:v for k,v in dict(
+    override_attrs = {
+        k: v
+        for k, v in dict(
             key_column=key_column,
             update_column=update_column,
             extra_columns=extra_columns,
             min_key=min_key,
             max_key=max_key,
             min_update=min_update,
             max_update=max_update,
-    ).items() if v is not None}
+        ).items()
+        if v is not None
+    }
 
     segments = [t.new(**override_attrs) for t in tables] if override_attrs else tables
 

data_diff/diff_tables.py

@@ -388,9 +388,16 @@ def diff_tables(self, table1: TableSegment, table2: TableSegment) -> DiffResult:
                 table1_count = self.stats.get("table1_count")
                 table2_count = self.stats.get("table2_count")
                 diff_count = self.stats.get("diff_count")
-                err_message = str(error)[:20]   # Truncate possibly sensitive information.
+                err_message = str(error)[:20]  # Truncate possibly sensitive information.
                 event_json = create_end_event_json(
-                    error is None, runtime, table1.database.name, table2.database.name, table1_count, table2_count, diff_count, err_message
+                    error is None,
+                    runtime,
+                    table1.database.name,
+                    table2.database.name,
+                    table1_count,
+                    table2_count,
+                    diff_count,
+                    err_message,
                 )
                 send_event_json(event_json)
 

tests/test_cli.py

@@ -12,7 +12,9 @@
 
 def run_datadiff_cli(*args):
     try:
-        stdout = subprocess.check_output([sys.executable, "-m", "data_diff", '--no-tracking'] + list(args), stderr=subprocess.PIPE)
+        stdout = subprocess.check_output(
+            [sys.executable, "-m", "data_diff", "--no-tracking"] + list(args), stderr=subprocess.PIPE
+        )
     except subprocess.CalledProcessError as e:
         logging.error(e.stderr)
         raise