update signature

sfu-db · Jan 23, 2025 · 8853a8a · 8853a8a
1 parent 71b8761
commit 8853a8a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/connectorx-python/src/lib.rs b/connectorx-python/src/lib.rs
@@ -40,7 +40,7 @@ fn connectorx(_: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
 }
 
 #[pyfunction]
-#[pyo3(signature = (conn, return_type, protocol=None, queries=None, partition_query=None))]
+#[pyo3(signature = (conn, return_type, protocol=None, queries=None, partition_query=None, pre_execution_queries=None))]
 pub fn read_sql<'py>(
     py: Python<'py>,
     conn: &str,
@@ -50,7 +50,15 @@ pub fn read_sql<'py>(
     partition_query: Option<cx_read_sql::PyPartitionQuery>,
     pre_execution_queries: Option<Vec<String>>,
 ) -> PyResult<Bound<'py, PyAny>> {
-    cx_read_sql::read_sql(py, conn, return_type, protocol, queries, partition_query, pre_execution_queries)
+    cx_read_sql::read_sql(
+        py,
+        conn,
+        return_type,
+        protocol,
+        queries,
+        partition_query,
+        pre_execution_queries,
+    )
 }
 
 #[pyfunction]