Skip to content

Commit 31139ee

Browse files
committed
do not use open_client to get version
1 parent bfbd52c commit 31139ee

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

pyiceberg/catalog/hive.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -407,12 +407,12 @@ def _create_hive_table(self, open_client: Client, hive_table: HiveTable) -> None
407407
raise TableAlreadyExistsError(f"Table {hive_table.dbName}.{hive_table.tableName} already exists") from e
408408

409409
def _get_hive_table(self, open_client: Client, *, dbname: str, tbl_name: str) -> HiveTable:
410-
if open_client._hive_version < 4:
410+
if self._client._hive_version < 4:
411411
return open_client.get_table(dbname=dbname, tbl_name=tbl_name)
412412
return open_client.get_table_req(GetTableRequest(dbName=dbname, tblName=tbl_name)).table
413413

414414
def _get_table_objects_by_name(self, open_client: Client, *, dbname: str, tbl_names: list[str]) -> list[HiveTable]:
415-
if open_client._hive_version < 4:
415+
if self._client._hive_version < 4:
416416
return open_client.get_table_objects_by_name(dbname=dbname, tbl_names=tbl_names)
417417
return open_client.get_table_objects_by_name_req(GetTablesRequest(dbName=dbname, tblNames=tbl_names)).tables
418418

tests/catalog/test_hive.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -955,7 +955,7 @@ def test_rename_table_from_does_not_exists() -> None:
955955
catalog = HiveCatalog(HIVE_CATALOG_NAME, uri=HIVE_METASTORE_FAKE_URL)
956956

957957
catalog._client = MagicMock()
958-
catalog._client.__enter__()._hive_version = 3
958+
catalog._client._hive_version = 3
959959
catalog._client.__enter__().alter_table_with_environment_context.side_effect = NoSuchObjectException(
960960
message="hive.default.does_not_exists table not found"
961961
)
@@ -970,7 +970,7 @@ def test_rename_table_to_namespace_does_not_exists() -> None:
970970
catalog = HiveCatalog(HIVE_CATALOG_NAME, uri=HIVE_METASTORE_FAKE_URL)
971971

972972
catalog._client = MagicMock()
973-
catalog._client.__enter__()._hive_version = 3
973+
catalog._client._hive_version = 3
974974
catalog._client.__enter__().alter_table_with_environment_context.side_effect = InvalidOperationException(
975975
message="Unable to change partition or table. Database default does not exist Check metastore logs for detailed stack.does_not_exists"
976976
)

0 commit comments

Comments
 (0)