fix: Row validation optimization to avoid select all columns

GoogleCloudPlatform · Oct 6, 2022 · 5a79223 · 5a79223
1 parent 9ebcae9
commit 5a79223
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/data_validation/query_builder/query_builder.py b/data_validation/query_builder/query_builder.py
@@ -488,10 +488,14 @@ def compile(self, validation_type, data_client, schema_name, table_name):
                 )
 
         if validation_type == consts.ROW_VALIDATION:
-            calc_table = calc_table.projection(self.compile_comparison_fields(calc_table))
+            calc_table = calc_table.projection(
+                self.compile_comparison_fields(calc_table)
+            )
         else:
             if self.comparison_fields:
-                calc_table = calc_table.mutate(self.compile_comparison_fields(calc_table))
+                calc_table = calc_table.mutate(
+                    self.compile_comparison_fields(calc_table)
+                )
         compiled_filters = self.compile_filter_fields(calc_table)
         filtered_table = (
             calc_table.filter(compiled_filters) if compiled_filters else calc_table