[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
sodadata · Oct 14, 2024 · 425ee8d · 425ee8d
1 parent 609949d
commit 425ee8d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/soda/core/soda/execution/check/row_count_comparison_check.py b/soda/core/soda/execution/check/row_count_comparison_check.py
@@ -41,8 +41,12 @@ def __init__(
 
  # If the other partition is None, we ignore the partition_cfg setup
  if row_count_comparison_check_cfg.other_partition_name:
- other_table_cfg = data_source_scan_cfg.get_or_create_table_cfg(row_count_comparison_check_cfg.other_table_name)
- other_partition_cfg = other_table_cfg.find_partition(row_count_comparison_check_cfg.location.file_path, row_count_comparison_check_cfg.other_partition_name)
+ other_table_cfg = data_source_scan_cfg.get_or_create_table_cfg(
+ row_count_comparison_check_cfg.other_table_name
+ )
+ other_partition_cfg = other_table_cfg.find_partition(
+ row_count_comparison_check_cfg.location.file_path, row_count_comparison_check_cfg.other_partition_name
+ )
  self.other_partition.set_partition_cfg(other_partition_cfg)
 
  self.metrics["row_count"] = self.data_source_scan.resolve_metric(

diff --git a/soda/core/tests/data_source/test_row_count_comparison.py b/soda/core/tests/data_source/test_row_count_comparison.py
@@ -62,6 +62,7 @@ def test_row_count_comparison_cross_data_source(data_source_fixture: DataSourceF
 
  scan.assert_all_checks_pass()
 
+
 def test_row_count_comparison_cross_data_source_with_filter(data_source_fixture: DataSourceFixture):
  """Does not really create two connections and test cross data sources with filtering, that is handled in integration tests.