[BatchAI] Remove implicit dependency on VM command module (Azure#7015)

BandaruDheeraj · Aug 10, 2018 · a9767ba · a9767ba
1 parent 6830ba2
commit a9767ba
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/src/command_modules/azure-cli-batchai/HISTORY.rst b/src/command_modules/azure-cli-batchai/HISTORY.rst
@@ -3,6 +3,10 @@
 Release History
 ===============
 
+0.4.2
++++++
+* Minor fixes
+
 0.4.1
 +++++
 * Logger output for auto-storage account creation now specifies "resource *group*".

diff --git a/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/_params.py b/src/command_modules/azure-cli-batchai/azure/cli/command_modules/batchai/_params.py
@@ -4,12 +4,18 @@
 # --------------------------------------------------------------------------------------------
 from argcomplete import FilesCompleter
 from azure.cli.command_modules.batchai import custom
-from azure.cli.command_modules.vm._actions import get_vm_sizes
 from azure.cli.core.commands.parameters import (
     get_enum_type, get_one_of_subscription_locations, resource_group_name_type, get_location_type)
 from azure.cli.core.decorators import Completer
 
 
+def get_vm_sizes(cli_ctx, location):
+    from azure.cli.core.profiles import ResourceType
+    from azure.cli.core.commands.client_factory import get_mgmt_service_client
+    client_factory = get_mgmt_service_client(cli_ctx, ResourceType.MGMT_COMPUTE)
+    return list(client_factory.virtual_machine_sizes.list(location))
+
+
 @Completer
 def get_vm_size_completion_list(cmd, prefix, namespace, **kwargs):  # pylint: disable=unused-argument
     try:

diff --git a/src/command_modules/azure-cli-batchai/setup.py b/src/command_modules/azure-cli-batchai/setup.py
@@ -13,7 +13,7 @@
     logger.warn("Wheel is not available, disabling bdist_wheel hook")
     cmdclass = {}
 
-VERSION = "0.4.1"
+VERSION = "0.4.2"
 # The full list of classifiers is available at
 # https://pypi.python.org/pypi?%3Aaction=list_classifiers
 CLASSIFIERS = [