change to type checking (#5062)

langchain-ai · May 21, 2023 · 8c661ba · 8c661ba
1 parent 424a573
commit 8c661ba
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/langchain/utilities/powerbi.py b/langchain/utilities/powerbi.py
@@ -4,7 +4,7 @@
 import logging
 import os
 from copy import deepcopy
-from typing import Any, Dict, Iterable, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Union
 
 import aiohttp
 import requests
@@ -16,13 +16,8 @@
 
 BASE_URL = os.getenv("POWERBI_BASE_URL", "https://api.powerbi.com/v1.0/myorg")
 
-try:
+if TYPE_CHECKING:
     from azure.core.credentials import TokenCredential
-except ImportError:
-    _LOGGER.log(
-        logging.WARNING,
-        "Could not import azure.core python package.",
-    )
 
 
 class PowerBIDataset(BaseModel):