Fixed the AccessToken type hint

craigktreasure · Feb 8, 2021 · 2d13e26 · 2d13e26
1 parent d86e3fa
commit 2d13e26
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/...xedreality/azure-mixedreality-authentication/azure/mixedreality/authentication/_client.py b/...xedreality/azure-mixedreality-authentication/azure/mixedreality/authentication/_client.py
@@ -81,9 +81,11 @@ def __init__(self, account_id, account_domain, credential, **kwargs):
 
     @distributed_trace
     def get_token(self, **kwargs):
-        # type: (Any) -> AccessToken
+        # type: (Any) -> azure.core.credentials.AccessToken
         """
         Retrieve a token from the STS service for the specified account identifier asynchronously.
+        :return: Instance of azure.core.credentials.AccessToken - token and expiry date of it
+        :rtype: ~azure.core.credentials.AccessToken
         """
         token_request_options = TokenRequestOptions()
         token_request_options.client_request_id = _generate_cv_base()

diff --git a/.../azure-mixedreality-authentication/azure/mixedreality/authentication/aio/_client_async.py b/.../azure-mixedreality-authentication/azure/mixedreality/authentication/aio/_client_async.py
@@ -13,7 +13,7 @@
 # pylint: disable=unused-import,ungrouped-imports
 from typing import Any, Union
 
-from azure.core.credentials import AccessToken, AzureKeyCredential
+from azure.core.credentials import AzureKeyCredential
 from azure.core.tracing.decorator_async import distributed_trace_async
 from azure.core.pipeline.policies import AsyncBearerTokenCredentialPolicy
 
@@ -25,6 +25,7 @@
 from ..utils import _convert_to_access_token, _generate_cv_base
 
 if TYPE_CHECKING:
+    from azure.core.credentials import AccessToken
     from azure.core.credentials_async import AsyncTokenCredential
 
 
@@ -84,9 +85,11 @@ def __init__(self,
             **kwargs)
 
     @distributed_trace_async
-    async def get_token(self, **kwargs) -> AccessToken:
+    async def get_token(self, **kwargs) -> "AccessToken":
         """
         Retrieve a token from the STS service for the specified account identifier asynchronously.
+        :return: Instance of azure.core.credentials.AccessToken - token and expiry date of it
+        :rtype: :class:`azure.core.credentials.AccessToken`
         """
         token_request_options = TokenRequestOptions()
         token_request_options.client_request_id=_generate_cv_base()