udpate transport

Author: igorvoltaic

pyiceberg/catalog/hive.py

@@ -157,7 +157,7 @@
 class _HiveClient:
     """Helper class to nicely open and close the transport."""
 
-    _transport: TTransport.TTransportBase
+    _transport: TTransport
     _ugi: Optional[List[str]]
     _hive_version: HiveVersion = HiveVersion(4, 0, 0)
     _hms_v3: object
@@ -184,7 +184,7 @@ def _get_hive_version(self) -> HiveVersion:
             version = map(int, open_client.getVersion().split("."))
             return HiveVersion(*version)
 
-    def _init_thrift_transport(self) -> TTransport.TTransportBase:
+    def _init_thrift_transport(self) -> TTransport:
         url_parts = urlparse(self._uri)
         if not url_parts.hostname or not url_parts.port:
             raise ValueError("hive hostname and port must be set")