[load] limit the scope of run_az_cmd import

src/load/HISTORY.rst

@@ -2,6 +2,11 @@
 
 Release History
 ===============
+1.4.1
+++++++
+* Move the `from azure.cli.core.util import run_az_cmd` to function scope to unblock users of existing features and most of the new features. Only the high scale download files feature requires this import, and it will be fixed in the next release.
+
+
 1.4.0
 ++++++
 * Add support for multi-region load test configuration. Multi-region load test configuration can be set using `--regionwise-engines` argument in 'az load test create' and 'az load test update' commands. Multi-region load test configuration set in YAML config file under key `regionalLoadTestConfig` will also be honoured.

src/load/azext_load/data_plane/utils/utils.py

@@ -18,7 +18,6 @@
     CLIInternalError,
 )
 from azure.mgmt.core.tools import is_valid_resource_id, parse_resource_id
-from azure.cli.core.util import run_az_cmd
 from knack.log import get_logger
 
 from .models import IdentityType, AllowedFileTypes, AllowedTestTypes
@@ -181,6 +180,7 @@ def download_file(url, file_path):
 
 
 def download_from_storage_container(sas_url, path):
+    from azure.cli.core.util import run_az_cmd
     logger.debug("Downloading files from storage container")
     cmd = ["az", "storage", "copy", "--source", sas_url, "--destination", path, "--recursive"]
     logger.debug("Executing command: %s", cmd)

src/load/setup.py

@@ -10,7 +10,7 @@
 
 
 # HISTORY.rst entry.
-VERSION = '1.4.0'
+VERSION = '1.4.1'
 
 # The full list of classifiers is available at
 # https://pypi.python.org/pypi?%3Aaction=list_classifiers