add doc argument in special operators

Author: finn-rudolph

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")

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,
         )
 
 

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")

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")

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,
         )