Fixed pyright error for MsalCredential (#35415)

* Fixed pyright error for `MsalCredential`

* update

* updates

* Update sdk/identity/azure-identity/CHANGELOG.md

Co-authored-by: Paul Van Eck <[email protected]>

* Update sdk/identity/azure-identity/azure/identity/_internal/msal_credentials.py

Co-authored-by: Paul Van Eck <[email protected]>

---------

Co-authored-by: Paul Van Eck <[email protected]>

Author: xiangyan99

sdk/identity/azure-identity/CHANGELOG.md

@@ -8,6 +8,8 @@
 
 ### Bugs Fixed
 
+- Fixed typing errors when certain credentials are used as context managers. ([#35415](https://github.com/Azure/azure-sdk-for-python/pull/35415))
+
 ### Other Changes
 
 ## 1.16.0 (2024-04-09)

sdk/identity/azure-identity/azure/identity/_internal/msal_credentials.py

@@ -4,6 +4,7 @@
 # ------------------------------------
 import os
 from typing import Any, List, Union, Dict, Optional
+from typing_extensions import Self
 
 import msal
 
@@ -59,7 +60,7 @@ def __init__(
 
         super(MsalCredential, self).__init__()
 
-    def __enter__(self) -> "MsalCredential":
+    def __enter__(self) -> Self:
         self._client.__enter__()
         return self
 

sdk/identity/azure-identity/setup.py

@@ -63,5 +63,6 @@
         "cryptography>=2.5",
         "msal>=1.24.0",
         "msal-extensions>=0.3.0",
+        "typing-extensions>=4.0.0",
     ],
 )