From 9940985a7bfa00b33be320e5be52da1e6cc9ae56 Mon Sep 17 00:00:00 2001 From: Zunli Hu Date: Sat, 9 May 2020 11:40:17 +0800 Subject: [PATCH 1/2] batchai cluster file --- src/azure-cli/azure/cli/command_modules/batchai/custom.py | 3 +++ 1 file changed, 3 insertions(+) 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 9d93c6dc99a..01fa7143411 100644 --- a/src/azure-cli/azure/cli/command_modules/batchai/custom.py +++ b/src/azure-cli/azure/cli/command_modules/batchai/custom.py @@ -766,6 +766,9 @@ def _get_files_from_bfs(cli_ctx, bfs, path, expiry): """ from azure.storage.blob import BlockBlobService from azure.storage.blob.models import Blob, BlobPermissions + BlockBlobService = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#BlockBlobService') + Blob = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#Blob') + BlobPermissions = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#BlobPermissions') result = [] service = BlockBlobService(bfs.account_name, _get_storage_account_key(cli_ctx, bfs.account_name, None)) effective_path = _get_path_for_storage(path) From af1bbf5f9d58279aff30d51a40e66086b39b75fa Mon Sep 17 00:00:00 2001 From: Zunli Hu Date: Sat, 9 May 2020 11:43:16 +0800 Subject: [PATCH 2/2] remove unused --- src/azure-cli/azure/cli/command_modules/batchai/custom.py | 2 -- 1 file changed, 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 01fa7143411..340fff735e8 100644 --- a/src/azure-cli/azure/cli/command_modules/batchai/custom.py +++ b/src/azure-cli/azure/cli/command_modules/batchai/custom.py @@ -764,8 +764,6 @@ def _get_files_from_bfs(cli_ctx, bfs, path, expiry): :param str path: path to list files from. :param int expiry: SAS expiration time in minutes. """ - from azure.storage.blob import BlockBlobService - from azure.storage.blob.models import Blob, BlobPermissions BlockBlobService = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#BlockBlobService') Blob = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#Blob') BlobPermissions = get_sdk(cli_ctx, ResourceType.DATA_STORAGE, 'blob#BlobPermissions')