correct tests

Sage-Bionetworks · Jun 21, 2024 · 4f5bff2 · 4f5bff2
1 parent 90f07b4
commit 4f5bff2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/glue/jobs/compare_parquet_datasets.py b/src/glue/jobs/compare_parquet_datasets.py
@@ -691,7 +691,7 @@ def compare_datasets_by_data_type(
         s3_filesystem=s3_filesystem,
     )
     main_dataset = get_parquet_dataset(
-        filter=filter_values,
+        filter_values=filter_values,
         dataset_key=get_parquet_dataset_s3_path(
             parquet_bucket, main_namespace, data_type
         ),
@@ -818,7 +818,7 @@ def main():
         s3=s3,
         reports=reports,
         data_type=data_type,
-        parquet_bucket_name=args["parquet_bucket"],
+        parquet_bucket=args["parquet_bucket"],
         staging_namespace=args["staging_namespace"],
     )
     return

diff --git a/tests/test_compare_parquet_datasets.py b/tests/test_compare_parquet_datasets.py
@@ -1029,7 +1029,7 @@ def test_that_compare_datasets_by_data_type_calls_compare_datasets_by_data_type_
                     s3_filesystem=None,
                 ),
                 mock.call(
-                    filter="some_filter",
+                    filter_values="some_filter",
                     dataset_key=f"s3://{parquet_bucket_name}/main/parquet/dataset_fitbitactivitylogs",
                     s3_filesystem=None,
                 ),