Fixing failed tests

Author: singankit

sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/_azure/_token_manager.py

@@ -71,7 +71,7 @@ def get_aad_credential(self) -> Union[DefaultAzureCredential, ManagedIdentityCre
             # Fall back to using the parent implementation
             return super().get_aad_credential()
 
-    def get_token(self, *scopes: str, claims: str | None = None, tenant_id: str | None = None, enable_cae: bool = False, **kwargs: Any) -> AccessToken:
+    def get_token(self, *scopes: str, claims: Union[str, None] = None, tenant_id: Union[str, None] = None, enable_cae: bool = False, **kwargs: Any) -> AccessToken:
         """Get the API token. If the token is not available or has expired, refresh the token.
 
         :return: API token

sdk/evaluation/azure-ai-evaluation/tests/e2etests/test_lite_management_client.py

@@ -1,7 +1,7 @@
 from typing import Any, Mapping
 import pytest
 import logging
-from azure.core.credentials import AzureSasCredential, TokenCredential
+from azure.core.credentials import AzureSasCredential, TokenCredential, AccessToken
 from azure.ai.evaluation._azure._clients import LiteMLClient
 
 
@@ -31,7 +31,7 @@ def test_get_token(self, project_scope, azure_cred):
         )
 
         token = client.get_token()
-        assert isinstance(token, str) and len(token) > 0
+        assert isinstance(token, AccessToken) and len(token) > 0
 
     @pytest.mark.azuretest
     @pytest.mark.parametrize("include_credentials", [False, True])