#2307

mabel-dev · Jan 25, 2025 · 1fe28e3 · 1fe28e3
1 parent 03364d1
commit 1fe28e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/opteryx/operators/aggregate_node.py b/opteryx/operators/aggregate_node.py
@@ -103,9 +103,11 @@ def build_aggregations(aggregators):
                 count_options = pyarrow.compute.CountOptions(mode="all")
             else:
                 field_name = field_node.schema_column.identity
+            if aggregator.value == "COUNT" and aggregator.duplicate_treatment == "Distinct":
+                aggregator.value = "COUNT_DISTINCT"
             function = AGGREGATORS[aggregator.value]
             # if the array agg is distinct, base off that function instead
-            if aggregator.value == "ARRAY_AGG" and aggregator.duplicate_treatment == "Distinct":
+             if aggregator.value == "ARRAY_AGG" and aggregator.duplicate_treatment == "Distinct":
                 function = "distinct"
             aggs.append((field_name, function, count_options))
             column_map[aggregator.schema_column.identity] = f"{field_name}_{function}".replace(