From 10d11c049073749f78816f327fa789c1733e02b5 Mon Sep 17 00:00:00 2001 From: Zunli Hu Date: Wed, 21 Apr 2021 13:27:51 +0800 Subject: [PATCH] refine client --- src/azure-cli/azure/cli/command_modules/batchai/custom.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/azure-cli/azure/cli/command_modules/batchai/custom.py b/src/azure-cli/azure/cli/command_modules/batchai/custom.py index 2b8a848dda0..b4c1e59d2ec 100644 --- a/src/azure-cli/azure/cli/command_modules/batchai/custom.py +++ b/src/azure-cli/azure/cli/command_modules/batchai/custom.py @@ -143,8 +143,8 @@ def _ensure_subnet_is_valid(client, subnet, nfs_resource_group, nfs_workspace, n def _get_storage_management_client(cli_ctx): - from azure.mgmt.storage import StorageManagementClient - return get_mgmt_service_client(cli_ctx, StorageManagementClient) + from azure.cli.core.profiles._shared import ResourceType + return get_mgmt_service_client(cli_ctx, ResourceType.MGMT_STORAGE) def _get_storage_account_key(cli_ctx, account_name, account_key):