Skip to content

Commit 03c5f78

Browse files
committed
fix type
1 parent 016adc0 commit 03c5f78

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

connectorx-python/connectorx/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,7 @@ def read_sql(
442442
return df
443443

444444

445-
def reconstruct_arrow_rb(results) -> Iterator[pa.RecordBatch]:
445+
def reconstruct_arrow_rb(results) -> pa.RecordBatchReader:
446446
import pyarrow as pa
447447

448448
# Get Schema

connectorx-python/src/arrow.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@ impl PyRecordBatch {
4343

4444
#[pymethods]
4545
impl PyRecordBatchIterator {
46-
4746
#[throws(ConnectorXPythonError)]
4847
fn schema_ptr<'py>(&self, py: Python<'py>) -> Bound<'py, PyAny> {
4948
let (rb, _) = self.0.get_schema();

0 commit comments

Comments
 (0)