Update auth.py

Azure · Mar 24, 2022 · 721bbc7 · 721bbc7
1 parent 37a27e0
commit 721bbc7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdk/cosmos/azure-cosmos/azure/cosmos/auth.py b/sdk/cosmos/azure-cosmos/azure/cosmos/auth.py
@@ -31,6 +31,7 @@
 from azure.core.exceptions import ServiceRequestError
 from azure.core.credentials import AccessToken
 from azure.core.pipeline import PipelineRequest, PipelineResponse
+from azure.core.pipeline.policies import HTTPPolicy
 from . import http_constants
 
 TokenCredential = ClassVar
@@ -203,7 +204,7 @@ def _need_new_token(self):
         return not self._token or self._token.expires_on - time.time() < 300
 
 
-class CosmosBearerTokenCredentialPolicy(_CosmosBearerTokenCredentialPolicyBase):
+class CosmosBearerTokenCredentialPolicy(_CosmosBearerTokenCredentialPolicyBase, HTTPPolicy):
     """Adds a Cosmos bearer token Authorization header to requests.
 
     :param credential: The credential.