azureml-registry-tools 0.1.0a8__tar.gz → 0.1.0a10__tar.gz

{azureml_registry_tools-0.1.0a8/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a8
+Version: 0.1.0a10
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license
@@ -10,7 +10,7 @@ Requires-Dist: azure-identity<2.0
 Requires-Dist: ruamel-yaml<0.19,>=0.17.21
 Requires-Dist: diskcache~=5.6
 Requires-Dist: azure-ai-ml<2.0
-Requires-Dist: azureml-assets<2.0
+Requires-Dist: azureml-assets<=1.16.86
 Dynamic: author
 Dynamic: license
 Dynamic: license-file

{azureml_registry_tools-0.1.0a8 → azureml_registry_tools-0.1.0a10}/azureml/registry/mgmt/asset_management.py

@@ -107,12 +107,13 @@ def set_storage_and_sas(asset: AssetConfig, storage_config: dict):
     _ = extra_config.path.get_uri(token_expiration=timedelta(hours=1))
 
 
-def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str) -> AssetConfig:
+def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, storage_overrides_exist: bool = False) -> AssetConfig:
     """Build a mutable copy of the asset in a temporary directory.
 
     Args:
         base_asset (AssetConfig): Base asset configuration to copy
         mutable_asset_dir (str): Directory path for the mutable asset copy
+        storage_overrides_exist (bool, optional): If True, model config will be modified to set type to custom_model.
 
     Returns:
         AssetConfig: Mutable asset configuration object
@@ -124,12 +125,14 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str) -> Asse
     mutable_asset_dir = Path(mutable_asset_dir).resolve()
     base_asset_file = base_asset.file_name_with_path.resolve()
     base_spec_file = base_asset.spec_with_path.resolve()
+    base_model_file = base_asset.extra_config_with_path.resolve()
 
     shutil.copytree(common_dir, mutable_asset_dir, dirs_exist_ok=True)
 
     # Reference asset files in mutable directory
     asset_config_file = mutable_asset_dir / base_asset_file.relative_to(common_dir)
     spec_config_file = mutable_asset_dir / base_spec_file.relative_to(common_dir)
+    model_config_file = mutable_asset_dir / base_model_file.relative_to(common_dir)
 
     # Autoincrement version for mutable asset
     with open(spec_config_file, "r") as f:
@@ -139,6 +142,16 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str) -> Asse
     with open(spec_config_file, "w") as f:
         yaml.dump(spec_config, f)
 
+    # If storage overrides are provided, default set model type to custom_model
+    if storage_overrides_exist:
+        print("Storage overrides provided, default setting model type to custom_model")
+        with open(model_config_file, "r") as f:
+            model_config = yaml.safe_load(f)
+            model_config["publish"]["type"] = "custom_model"
+
+        with open(model_config_file, "w") as f:
+            yaml.dump(model_config, f)
+
     mutable_asset = AssetConfig(asset_config_file)
 
     return mutable_asset
@@ -162,7 +175,7 @@ def create_or_update_asset(readonly_asset: AssetConfig, config: RegistryConfig):
     )
 
     with tempfile.TemporaryDirectory() as mutable_asset_dir:
-        mutable_asset = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir)
+        mutable_asset = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir, storage_overrides_exist=bool(config.storage_config))
         # autoincrement version
         try:
             set_storage_and_sas(mutable_asset, config.storage_config)

{azureml_registry_tools-0.1.0a8 → azureml_registry_tools-0.1.0a10/azureml_registry_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a8
+Version: 0.1.0a10
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license
@@ -10,7 +10,7 @@ Requires-Dist: azure-identity<2.0
 Requires-Dist: ruamel-yaml<0.19,>=0.17.21
 Requires-Dist: diskcache~=5.6
 Requires-Dist: azure-ai-ml<2.0
-Requires-Dist: azureml-assets<2.0
+Requires-Dist: azureml-assets<=1.16.86
 Dynamic: author
 Dynamic: license
 Dynamic: license-file

{azureml_registry_tools-0.1.0a8 → azureml_registry_tools-0.1.0a10}/azureml_registry_tools.egg-info/requires.txt

@@ -2,4 +2,4 @@ azure-identity<2.0
 ruamel-yaml<0.19,>=0.17.21
 diskcache~=5.6
 azure-ai-ml<2.0
-azureml-assets<2.0
+azureml-assets<=1.16.86

{azureml_registry_tools-0.1.0a8 → azureml_registry_tools-0.1.0a10}/setup.py

@@ -11,14 +11,14 @@ DEPENDENCIES = [
     "ruamel-yaml>=0.17.21,<0.19",
     "diskcache~=5.6",
     "azure-ai-ml<2.0",
-    "azureml-assets<2.0"
+    "azureml-assets<=1.16.86"
 ]
 
 exclude_list = ["*.tests"]
 
 setup(
     name='azureml-registry-tools',
-    version="0.1.0a8",
+    version="0.1.0a10",
     description='AzureML Registry tools and CLI',
     author='Microsoft Corp',
     license="https://aka.ms/azureml-sdk-license",