aem: adjust command authoring code per cli core change (#85)

Azure · Feb 28, 2018 · d2ce8ef · d2ce8ef
1 parent d7d4dc1
commit d2ce8ef
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/aem/azext_aem/__init__.py b/src/aem/azext_aem/__init__.py
@@ -4,6 +4,7 @@
 # --------------------------------------------------------------------------------------------
 
 from azure.cli.core import AzCommandsLoader
+from azure.cli.core.profiles import ResourceType
 
 import azext_aem._help  # pylint: disable=unused-import
 
@@ -18,7 +19,7 @@ def __init__(self, cli_ctx=None):
                                                 custom_command_type=aem_custom)
 
     def load_command_table(self, _):
-        with self.command_group('vm aem', min_api='2016-04-30-preview') as g:
+        with self.command_group('vm aem', min_api='2016-04-30-preview', resource_type=ResourceType.MGMT_COMPUTE) as g:
             g.custom_command('set', 'set_aem')
             g.custom_command('delete', 'delete_aem')
             g.custom_command('verify', 'verify_aem')