Skip to content

Commit 46097d1

Browse files
committed
refactor: Update volatility parameter type in udwf method signature to support Volatility enum
1 parent d29acf6 commit 46097d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/datafusion/udf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -632,7 +632,7 @@ def __call__(self, *args: Expr) -> Expr:
632632
def udwf(
633633
input_type: pa.DataType | list[pa.DataType],
634634
return_type: pa.DataType,
635-
volatility: str,
635+
volatility: Volatility | str,
636636
name: Optional[str] = None,
637637
) -> Callable[..., WindowUDF]: ...
638638

@@ -642,7 +642,7 @@ def udwf(
642642
func: Callable[[], WindowEvaluator],
643643
input_type: pa.DataType | list[pa.DataType],
644644
return_type: pa.DataType,
645-
volatility: str,
645+
volatility: Volatility | str,
646646
name: Optional[str] = None,
647647
) -> WindowUDF: ...
648648

0 commit comments

Comments
 (0)