azureml-registry-tools 0.1.0a33__tar.gz → 0.1.0a34__tar.gz

{azureml_registry_tools-0.1.0a33/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a33
+Version: 0.1.0a34
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license

{azureml_registry_tools-0.1.0a33 → azureml_registry_tools-0.1.0a34}/azureml/registry/mgmt/create_manifest.py

@@ -20,14 +20,6 @@ ALLOWED_ASSET_TYPES = {
     'environments', 'models', 'deployment-templates', 'components', 'datasets'
 }
 
-ASSET_TYPE_MAP = {
-    'models': 'Models',
-    'environments': 'Environments',
-    'deployment-templates': 'DeploymentTemplates',
-    'components': 'Components',
-    'datasets': 'Datasets',
-}
-
 
 def validate_manifest_data(data: dict, file_path: str) -> None:
     """Validate required root-level fields and allowed values for nested fields under 'assets'.

{azureml_registry_tools-0.1.0a33 → azureml_registry_tools-0.1.0a34}/azureml/registry/mgmt/syndication_manifest.py

@@ -7,6 +7,18 @@ from dataclasses import dataclass, field
 from typing import List
 from uuid import UUID
 
+# Asset type mapping from YAML keys to JSON/API keys
+ASSET_TYPE_MAP = {
+    'models': 'Models',
+    'environments': 'Environments',
+    'deployment-templates': 'DeploymentTemplates',
+    'components': 'Components',
+    'datasets': 'Datasets',
+}
+
+# Reverse mapping for deserializing API responses back to YAML keys
+ASSET_TYPE_REVERSE_MAP = {v: k for k, v in ASSET_TYPE_MAP.items()}
+
 
 def _norm_key(k):
     return k.replace('_', '').lower()
@@ -70,7 +82,13 @@ class SourceRegistry:
         asset_dict = {}
         for asset_type, asset_list in self.assets.items():
             if asset_list:
-                asset_dict[asset_type] = [
+                # Convert YAML asset type keys for API
+                if asset_type not in ASSET_TYPE_MAP:
+                    raise ValueError(
+                        f"Invalid asset type '{asset_type}'. Allowed types: {', '.join(ASSET_TYPE_MAP.keys())}"
+                    )
+                api_asset_type = ASSET_TYPE_MAP[asset_type]
+                asset_dict[api_asset_type] = [
                     {"Name": a.name} if a.name == ".*" else {"Name": a.name, "Version": a.version}
                     for a in asset_list
                 ]
@@ -86,7 +104,9 @@ class SourceRegistry:
         assets = {}
         assets_dict = _get_key(d, "Assets", "assets", normalize_keys=normalize_keys)
         for asset_type, asset_list in assets_dict.items():
-            assets[asset_type] = [Asset.from_dict(a, normalize_keys=normalize_keys) for a in asset_list]
+            # Normalize API keys back to YAML keys
+            normalized_asset_type = ASSET_TYPE_REVERSE_MAP.get(asset_type, asset_type)
+            assets[normalized_asset_type] = [Asset.from_dict(a, normalize_keys=normalize_keys) for a in asset_list]
         return SourceRegistry(
             registry_name=_get_key(d, "RegistryName", "registry_name", normalize_keys=normalize_keys),
             tenant_id=UUID(str(_get_key(d, "TenantId", "tenant_id", normalize_keys=normalize_keys))),

{azureml_registry_tools-0.1.0a33 → azureml_registry_tools-0.1.0a34/azureml_registry_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a33
+Version: 0.1.0a34
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license

{azureml_registry_tools-0.1.0a33 → azureml_registry_tools-0.1.0a34}/setup.py

@@ -18,7 +18,7 @@ exclude_list = ["*.tests"]
 
 setup(
     name='azureml-registry-tools',
-    version="0.1.0a33",
+    version="0.1.0a34",
     description='AzureML Registry tools and CLI',
     author='Microsoft Corp',
     license="https://aka.ms/azureml-sdk-license",