Skip to content

Commit 79e04fd

Browse files
peterschmidt85Andrey Cheptsov
andauthored
Fix Azure resource client import (#3988)
Co-authored-by: Andrey Cheptsov <andrey.cheptsov@github.com>
1 parent 740d871 commit 79e04fd

2 files changed

Lines changed: 10 additions & 4 deletions

File tree

src/dstack/_internal/core/backends/azure/configurator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from azure.core.credentials import TokenCredential
77
from azure.mgmt import msi as msi_mgmt
88
from azure.mgmt import network as network_mgmt
9-
from azure.mgmt import resource as resource_mgmt
109
from azure.mgmt import subscription as subscription_mgmt
1110
from azure.mgmt.network.models import (
1211
AddressSpace,
@@ -18,6 +17,7 @@
1817
Subnet,
1918
VirtualNetwork,
2019
)
20+
from azure.mgmt.resource import resources as resource_mgmt
2121
from azure.mgmt.resource.resources.models import ResourceGroup
2222

2323
from dstack._internal.core.backends.azure import auth, compute, resources

src/tests/_internal/server/routers/test_backends.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -492,9 +492,15 @@ async def test_create_azure_backend(self, test_db, session: AsyncSession, client
492492
}
493493
with (
494494
patch("dstack._internal.core.backends.azure.auth.authenticate") as authenticate_mock,
495-
patch("azure.mgmt.subscription.SubscriptionClient") as SubscriptionClientMock,
496-
patch("azure.mgmt.resource.ResourceManagementClient") as ResourceManagementClientMock,
497-
patch("azure.mgmt.network.NetworkManagementClient") as NetworkManagementClientMock,
495+
patch(
496+
"dstack._internal.core.backends.azure.configurator.subscription_mgmt.SubscriptionClient"
497+
) as SubscriptionClientMock,
498+
patch(
499+
"dstack._internal.core.backends.azure.configurator.resource_mgmt.ResourceManagementClient"
500+
) as ResourceManagementClientMock,
501+
patch(
502+
"dstack._internal.core.backends.azure.configurator.network_mgmt.NetworkManagementClient"
503+
) as NetworkManagementClientMock,
498504
):
499505
authenticate_mock.return_value = None, "test_tenant"
500506
subscription_client_mock = SubscriptionClientMock.return_value

0 commit comments

Comments
 (0)