From c382e0f16fc3a15fb5aac31cbd5957214e32f445 Mon Sep 17 00:00:00 2001
From: Andrey Cheptsov <andrey.cheptsov@github.com>
Date: Wed, 24 Jun 2026 22:26:52 +0200
Subject: [PATCH] Fix Azure resource client import

---
 .../_internal/core/backends/azure/configurator.py    |  2 +-
 src/tests/_internal/server/routers/test_backends.py  | 12 +++++++++---
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/dstack/_internal/core/backends/azure/configurator.py b/src/dstack/_internal/core/backends/azure/configurator.py
index d3cf0649d..31297e4ce 100644
--- a/src/dstack/_internal/core/backends/azure/configurator.py
+++ b/src/dstack/_internal/core/backends/azure/configurator.py
@@ -6,7 +6,6 @@
 from azure.core.credentials import TokenCredential
 from azure.mgmt import msi as msi_mgmt
 from azure.mgmt import network as network_mgmt
-from azure.mgmt import resource as resource_mgmt
 from azure.mgmt import subscription as subscription_mgmt
 from azure.mgmt.network.models import (
     AddressSpace,
@@ -18,6 +17,7 @@
     Subnet,
     VirtualNetwork,
 )
+from azure.mgmt.resource import resources as resource_mgmt
 from azure.mgmt.resource.resources.models import ResourceGroup
 
 from dstack._internal.core.backends.azure import auth, compute, resources
diff --git a/src/tests/_internal/server/routers/test_backends.py b/src/tests/_internal/server/routers/test_backends.py
index 79fb13667..62eff12a3 100644
--- a/src/tests/_internal/server/routers/test_backends.py
+++ b/src/tests/_internal/server/routers/test_backends.py
@@ -492,9 +492,15 @@ async def test_create_azure_backend(self, test_db, session: AsyncSession, client
         }
         with (
             patch("dstack._internal.core.backends.azure.auth.authenticate") as authenticate_mock,
-            patch("azure.mgmt.subscription.SubscriptionClient") as SubscriptionClientMock,
-            patch("azure.mgmt.resource.ResourceManagementClient") as ResourceManagementClientMock,
-            patch("azure.mgmt.network.NetworkManagementClient") as NetworkManagementClientMock,
+            patch(
+                "dstack._internal.core.backends.azure.configurator.subscription_mgmt.SubscriptionClient"
+            ) as SubscriptionClientMock,
+            patch(
+                "dstack._internal.core.backends.azure.configurator.resource_mgmt.ResourceManagementClient"
+            ) as ResourceManagementClientMock,
+            patch(
+                "dstack._internal.core.backends.azure.configurator.network_mgmt.NetworkManagementClient"
+            ) as NetworkManagementClientMock,
         ):
             authenticate_mock.return_value = None, "test_tenant"
             subscription_client_mock = SubscriptionClientMock.return_value