fix

apache · Dec 3, 2024 · 4e9a265 · 4e9a265
1 parent 1e823ed
commit 4e9a265
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pypaimon/py4j/java_implementation.py b/pypaimon/py4j/java_implementation.py
@@ -172,14 +172,14 @@ def to_duckdb(
             self,
             splits: List[Split],
             table_name: str,
-            connection: Optional[DuckDBPyConnection] = None) -> DuckDBPyConnection:
+            connection: Optional["DuckDBPyConnection"] = None) -> "DuckDBPyConnection":
         import duckdb
 
         con = connection or duckdb.connect(database=":memory:")
         con.register(table_name, self.to_arrow(splits))
         return con
 
-    def to_ray(self, splits: List[Split]) -> ray.data.dataset.Dataset:
+    def to_ray(self, splits: List[Split]) -> "ray.data.dataset.Dataset":
         import ray
 
         return ray.data.from_arrow(self.to_arrow(splits))