Small bugfix. When arguments are None, we should use count_star

timsaucer · Jul 24, 2024 · 5cabf83 · 5cabf83
1 parent f00b8ee
commit 5cabf83
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/datafusion/functions.py b/python/datafusion/functions.py
@@ -1294,6 +1294,8 @@ def corr(value1: Expr, value2: Expr, distinct: bool = False) -> Expr:
 
 def count(args: Expr | list[Expr] | None = None, distinct: bool = False) -> Expr:
     """Returns the number of rows that match the given arguments."""
+    if args is None:
+        return Expr(f.count_star())
     if isinstance(args, list):
         args = [arg.expr for arg in args]
     elif isinstance(args, Expr):