[Tables] updating docstring (Azure#19547)

Author: seankane-msft

sdk/tables/azure-data-tables/azure/data/tables/_base_client.py

@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 
-from typing import Dict, Optional, Any, List, Mapping, Union
+from typing import Dict, Optional, Any, List, Mapping, Union, TYPE_CHECKING
 from uuid import uuid4
 try:
     from urllib.parse import parse_qs, quote, urlparse
@@ -50,6 +50,8 @@
 )
 from ._sdk_moniker import SDK_MONIKER
 
+if TYPE_CHECKING:
+    from azure.core.credentials import TokenCredential
 
 _SUPPORTED_API_VERSIONS = ["2019-02-02", "2019-07-07"]
 
@@ -71,7 +73,7 @@ class AccountHostsMixin(object):  # pylint: disable=too-many-instance-attributes
     def __init__(
         self,
         account_url,  # type: Any
-        credential=None,  # type: Optional[Union[AzureNamedKeyCredential, AzureSasCredential]]
+        credential=None,  # type: Optional[Union[AzureNamedKeyCredential, AzureSasCredential, "TokenCredential"]]
         **kwargs  # type: Any
     ):
         # type: (...) -> None

sdk/tables/azure-data-tables/azure/data/tables/aio/_base_client_async.py

@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, List, Mapping, Optional, Union, TYPE_CHECKING
 from uuid import uuid4
 
 from azure.core.credentials import AzureSasCredential, AzureNamedKeyCredential
@@ -35,17 +35,20 @@
 from .._sdk_moniker import SDK_MONIKER
 from ._policies_async import AsyncTablesRetryPolicy
 
+if TYPE_CHECKING:
+    from azure.core.credentials_async import AsyncTokenCredential
+
 
 class AsyncTablesBaseClient(AccountHostsMixin):
 
     def __init__(  # pylint: disable=missing-client-constructor-parameter-credential
         self,
         endpoint: str,
         *,
-        credential: Optional[Union[AzureSasCredential, AzureNamedKeyCredential]] = None,
+        credential: Optional[Union[AzureSasCredential, AzureNamedKeyCredential, "AsyncTokenCredential"]] = None,
         **kwargs: Any
     ) -> None:
-        super(AsyncTablesBaseClient, self).__init__(endpoint, credential=credential, **kwargs)
+        super(AsyncTablesBaseClient, self).__init__(endpoint, credential=credential, **kwargs)  # type: ignore
         self._client = AzureTable(
             self.url,
             policies=kwargs.pop('policies', self._policies),

sdk/tables/azure-data-tables/azure/data/tables/aio/_table_client_async.py

@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 import functools
-from typing import List, Union, Any, Optional, Mapping, Iterable, Dict, overload, cast
+from typing import List, Union, Any, Optional, Mapping, Iterable, Dict, overload, cast, TYPE_CHECKING
 try:
     from urllib.parse import urlparse, unquote
 except ImportError:
@@ -38,6 +38,9 @@
 from ._models import TableEntityPropertiesPaged
 from ._table_batch_async import TableBatchOperations
 
+if TYPE_CHECKING:
+    from azure.core.credentials_async import AsyncTokenCredential
+
 
 class TableClient(AsyncTablesBaseClient):
     """A client to interact with a specific Table in an Azure Tables account.
@@ -52,7 +55,7 @@ def __init__(  # pylint: disable=missing-client-constructor-parameter-credential
         endpoint: str,
         table_name: str,
         *,
-        credential: Optional[Union[AzureSasCredential, AzureNamedKeyCredential]] = None,
+        credential: Optional[Union[AzureSasCredential, AzureNamedKeyCredential, "AsyncTokenCredential"]] = None,
         **kwargs
     ) -> None:
         """Create TableClient from a Credential.

sdk/tables/azure-data-tables/azure/data/tables/aio/_table_service_client_async.py

@@ -326,7 +326,7 @@ def get_table_client(self, table_name: str, **kwargs) -> TableClient:
         return TableClient(
             self.url,
             table_name=table_name,
-            credential=self.credential,
+            credential=self.credential,  # type: ignore
             api_version=self.api_version,
             pipeline=pipeline,
             location_mode=self._location_mode,