Skip to content

Commit 89fc0f6

Browse files
committed
fix type hint
1 parent 25ddb2f commit 89fc0f6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pyiceberg/table/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,7 @@
147147
import pyarrow as pa
148148
import ray
149149
from duckdb import DuckDBPyConnection
150+
from pyiceberg_core.datafusion import IcebergDataFusionTable
150151

151152
from pyiceberg.catalog import Catalog
152153

@@ -1428,7 +1429,7 @@ def to_polars(self) -> pl.LazyFrame:
14281429

14291430
return pl.scan_iceberg(self)
14301431

1431-
def __datafusion_table_provider__(self) -> Any:
1432+
def __datafusion_table_provider__(self) -> "IcebergDataFusionTable":
14321433
"""Return the DataFusion table provider PyCapsule interface.
14331434
14341435
To support DataFusion features such as push down filtering, this function will return a PyCapsule

0 commit comments

Comments
 (0)