Skip to content

Commit 15899b5

Browse files
committed
fix type hint
1 parent 299fc48 commit 15899b5

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
@@ -143,6 +143,7 @@
143143
import pyarrow as pa
144144
import ray
145145
from duckdb import DuckDBPyConnection
146+
from pyiceberg_core.datafusion import IcebergDataFusionTable
146147

147148
from pyiceberg.catalog import Catalog
148149

@@ -1494,7 +1495,7 @@ def to_polars(self) -> pl.LazyFrame:
14941495

14951496
return pl.scan_iceberg(self)
14961497

1497-
def __datafusion_table_provider__(self) -> Any:
1498+
def __datafusion_table_provider__(self) -> "IcebergDataFusionTable":
14981499
"""Return the DataFusion table provider PyCapsule interface.
14991500
15001501
To support DataFusion features such as push down filtering, this function will return a PyCapsule

0 commit comments

Comments
 (0)