diff --git a/src/pydiverse/transform/_internal/ops/ops/datetime.py b/src/pydiverse/transform/_internal/ops/ops/datetime.py index 8607269..194f3b0 100644 --- a/src/pydiverse/transform/_internal/ops/ops/datetime.py +++ b/src/pydiverse/transform/_internal/ops/ops/datetime.py @@ -6,16 +6,17 @@ class DatetimeExtract(Operator): - def __init__(self, name: str): - super().__init__(name, Signature(Datetime(), return_type=Int())) + def __init__(self, name: str, doc: str = ""): + super().__init__(name, Signature(Datetime(), return_type=Int()), doc=doc) class DateExtract(Operator): - def __init__(self, name: str): + def __init__(self, name: str, doc: str = ""): super().__init__( name, Signature(Date(), return_type=Int()), Signature(Datetime(), return_type=Int()), + doc=doc, ) @@ -41,8 +42,8 @@ def __init__(self, name: str): class DurationToUnit(Operator): - def __init__(self, name: str): - super().__init__(name, Signature(Duration(), return_type=Int())) + def __init__(self, name: str, doc: str = ""): + super().__init__(name, Signature(Duration(), return_type=Int()), doc=doc) dur_days = DurationToUnit("dur.days") diff --git a/src/pydiverse/transform/_internal/ops/ops/horizontal.py b/src/pydiverse/transform/_internal/ops/ops/horizontal.py index 559be56..6fb2b45 100644 --- a/src/pydiverse/transform/_internal/ops/ops/horizontal.py +++ b/src/pydiverse/transform/_internal/ops/ops/horizontal.py @@ -6,9 +6,13 @@ class Horizontal(Operator): - def __init__(self, name: str, *signatures: Signature): + def __init__(self, name: str, *signatures: Signature, doc: str = ""): super().__init__( - name, *signatures, param_names=["arg", "args"], generate_expr_method=False + name, + *signatures, + param_names=["arg", "args"], + generate_expr_method=False, + doc=doc, ) diff --git a/src/pydiverse/transform/_internal/ops/ops/markers.py b/src/pydiverse/transform/_internal/ops/ops/markers.py index ac6b8ae..3a94478 100644 --- a/src/pydiverse/transform/_internal/ops/ops/markers.py +++ b/src/pydiverse/transform/_internal/ops/ops/markers.py @@ -6,8 +6,8 @@ class Marker(Operator): - def __init__(self, name: str): - super().__init__(name, Signature(D, return_type=D)) + def __init__(self, name: str, doc: str = ""): + super().__init__(name, Signature(D, return_type=D), doc=doc) nulls_first = Marker("nulls_first") diff --git a/src/pydiverse/transform/_internal/ops/ops/string.py b/src/pydiverse/transform/_internal/ops/ops/string.py index d297af6..abca639 100644 --- a/src/pydiverse/transform/_internal/ops/ops/string.py +++ b/src/pydiverse/transform/_internal/ops/ops/string.py @@ -6,8 +6,8 @@ class StrUnary(Operator): - def __init__(self, name: str): - super().__init__(name, Signature(String(), return_type=String())) + def __init__(self, name: str, doc: str = ""): + super().__init__(name, Signature(String(), return_type=String()), doc=doc) str_strip = StrUnary("str.strip") diff --git a/src/pydiverse/transform/_internal/ops/ops/window.py b/src/pydiverse/transform/_internal/ops/ops/window.py index 0bef597..cff39e6 100644 --- a/src/pydiverse/transform/_internal/ops/ops/window.py +++ b/src/pydiverse/transform/_internal/ops/ops/window.py @@ -15,6 +15,7 @@ def __init__( param_names: list[str] | None = None, default_values: list[Any] | None = None, generate_expr_method=False, + doc: str = "", ): super().__init__( name, @@ -24,6 +25,7 @@ def __init__( param_names=param_names, default_values=default_values, generate_expr_method=generate_expr_method, + doc=doc, )