fix(python): default value for io threads (#8)

databendlabs · May 27, 2024 · 59edaf7 · 59edaf7
1 parent f2841d6
commit 59edaf7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/databend_udf/udf.py b/python/databend_udf/udf.py
@@ -61,7 +61,7 @@ class ScalarFunction(UserDefinedFunction):
     _skip_null: bool
 
     def __init__(
-        self, func, input_types, result_type, name=None, io_threads=32, skip_null=None
+        self, func, input_types, result_type, name=None, io_threads=None, skip_null=None
     ):
         self._func = func
         self._input_schema = pa.schema(
@@ -138,7 +138,7 @@ def udf(
     input_types: Union[List[Union[str, pa.DataType]], Union[str, pa.DataType]],
     result_type: Union[str, pa.DataType],
     name: Optional[str] = None,
-    io_threads: Optional[int] = None,
+    io_threads: Optional[int] = 32,
     skip_null: Optional[bool] = False,
 ) -> Callable:
     """

diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -7,7 +7,7 @@ classifiers = [
 description = "Databend UDF Server"
 license = { text = "Apache-2.0" }
 name = "databend-udf"
-version = "0.2.3"
+version = "0.2.4"
 readme = "README.md"
 requires-python = ">=3.7"
 dependencies = ["pyarrow"]