refactor(sdk): Use sdk instead of _sdk_extras (#13736)

This commit is contained in:
Anthony Burdi 2025-06-10 14:10:57 -04:00 committed by GitHub
parent 78cfc49703
commit eba14287bb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -10,7 +10,7 @@ from datahub.sdk.lineage_client import LineageClient
from datahub.sdk.search_client import SearchClient from datahub.sdk.search_client import SearchClient
try: try:
from acryl_datahub_cloud._sdk_extras import ( # type: ignore[import-not-found] from acryl_datahub_cloud.sdk import ( # type: ignore[import-not-found]
ResolverClient, ResolverClient,
) )
except ImportError: except ImportError:
@ -115,7 +115,7 @@ class DataHubClient:
@property @property
def assertions(self): # type: ignore[report-untyped-call] # Not available due to circular import issues def assertions(self): # type: ignore[report-untyped-call] # Not available due to circular import issues
try: try:
from acryl_datahub_cloud._sdk_extras import AssertionsClient from acryl_datahub_cloud.sdk import AssertionsClient
except ImportError as e: except ImportError as e:
if "acryl_datahub_cloud" in str(e): if "acryl_datahub_cloud" in str(e):
raise SdkUsageError( raise SdkUsageError(
@ -128,7 +128,7 @@ class DataHubClient:
@property @property
def subscriptions(self): # type: ignore[report-untyped-call] # Not available due to circular import issues def subscriptions(self): # type: ignore[report-untyped-call] # Not available due to circular import issues
try: try:
from acryl_datahub_cloud._sdk_extras import SubscriptionClient from acryl_datahub_cloud.sdk import SubscriptionClient
except ImportError as e: except ImportError as e:
if "acryl_datahub_cloud" in str(e): if "acryl_datahub_cloud" in str(e):
raise SdkUsageError( raise SdkUsageError(