From 8c661baefb9b1e05f787537459423171185c81a1 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 21 May 2023 09:09:49 -0700 Subject: [PATCH] change to type checking (#5062) --- langchain/utilities/powerbi.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/langchain/utilities/powerbi.py b/langchain/utilities/powerbi.py index 9af8cb66..f542c2d0 100644 --- a/langchain/utilities/powerbi.py +++ b/langchain/utilities/powerbi.py @@ -4,7 +4,7 @@ from __future__ import annotations 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 @@ _LOGGER = logging.getLogger(__name__) 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):