2222 write_builder , table_write , commit_message , table_commit )
2323from pyarrow import RecordBatchReader , RecordBatch
2424from typing import List
25- from typing_extensions import Self
2625
2726
2827class Catalog (catalog .Catalog ):
@@ -37,7 +36,7 @@ def create(catalog_context: dict) -> 'Catalog':
3736 j_catalog = gateway .jvm .CatalogFactory .createCatalog (j_catalog_context )
3837 return Catalog (j_catalog )
3938
40- def get_table (self , identifier : tuple ) -> 'Table' :
39+ def get_table (self , identifier : str ) -> 'Table' :
4140 gateway = get_gateway ()
4241 j_identifier = gateway .jvm .Identifier .fromString (identifier )
4342 j_table = self ._j_catalog .getTable (j_identifier )
@@ -63,11 +62,11 @@ class ReadBuilder(read_builder.ReadBuilder):
6362 def __init__ (self , j_read_builder ):
6463 self ._j_read_builder = j_read_builder
6564
66- def with_projection (self , projection : List [List [int ]]) -> Self :
65+ def with_projection (self , projection : List [List [int ]]) -> 'ReadBuilder' :
6766 self ._j_read_builder .withProjection (projection )
6867 return self
6968
70- def with_limit (self , limit : int ) -> Self :
69+ def with_limit (self , limit : int ) -> 'ReadBuilder' :
7170 self ._j_read_builder .withLimit (limit )
7271 return self
7372
@@ -121,7 +120,7 @@ class BatchWriteBuilder(write_builder.BatchWriteBuilder):
121120 def __init__ (self , j_batch_write_builder ):
122121 self ._j_batch_write_builder = j_batch_write_builder
123122
124- def with_overwrite (self , static_partition : dict ) -> Self :
123+ def with_overwrite (self , static_partition : dict ) -> 'BatchWriteBuilder' :
125124 self ._j_batch_write_builder .withOverwrite (static_partition )
126125 return self
127126
0 commit comments