From 4b86853219e43fcca3f3c96863f9460371eed18e Mon Sep 17 00:00:00 2001
From: Li Yazhou <me.ssword@gmail.com>
Date: Mon, 27 May 2024 12:57:54 +0800
Subject: [PATCH] fix the default value for io threads

---
 python/databend_udf/udf.py | 4 ++--
 python/pyproject.toml      | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/databend_udf/udf.py b/python/databend_udf/udf.py
index b577e59..6b20e72 100644
--- 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
index 95528c8..ddea18a 100644
--- 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"]