Skip to content

Commit 4e9a265

Browse files
author
yuzelin
committed
fix
1 parent 1e823ed commit 4e9a265

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pypaimon/py4j/java_implementation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -172,14 +172,14 @@ def to_duckdb(
172172
self,
173173
splits: List[Split],
174174
table_name: str,
175-
connection: Optional[DuckDBPyConnection] = None) -> DuckDBPyConnection:
175+
connection: Optional["DuckDBPyConnection"] = None) -> "DuckDBPyConnection":
176176
import duckdb
177177

178178
con = connection or duckdb.connect(database=":memory:")
179179
con.register(table_name, self.to_arrow(splits))
180180
return con
181181

182-
def to_ray(self, splits: List[Split]) -> ray.data.dataset.Dataset:
182+
def to_ray(self, splits: List[Split]) -> "ray.data.dataset.Dataset":
183183
import ray
184184

185185
return ray.data.from_arrow(self.to_arrow(splits))

0 commit comments

Comments
 (0)