From a6d624095401a4a5a3d29d78c005beeff80fe279 Mon Sep 17 00:00:00 2001 From: Colin Chartier Date: Mon, 24 Feb 2025 15:28:26 -0500 Subject: [PATCH] fix test --- .../query/processors/test_hash_bucket_functions_processor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/query/processors/test_hash_bucket_functions_processor.py b/tests/query/processors/test_hash_bucket_functions_processor.py index a259ca1066d..ea34879785d 100644 --- a/tests/query/processors/test_hash_bucket_functions_processor.py +++ b/tests/query/processors/test_hash_bucket_functions_processor.py @@ -220,7 +220,9 @@ @pytest.mark.parametrize("pre_format, expected_query", test_data) def test_format_expressions(pre_format: Query, expected_query: Query) -> None: copy = deepcopy(pre_format) - HashBucketFunctionTransformer("attr_str").process_query(copy, HTTPQuerySettings()) + HashBucketFunctionTransformer("attr_str", num_attribute_buckets=5).process_query( + copy, HTTPQuerySettings() + ) assert copy.get_selected_columns() == expected_query.get_selected_columns() assert copy.get_groupby() == expected_query.get_groupby() assert copy.get_condition() == expected_query.get_condition()