azureml-registry-tools 0.1.0a13__tar.gz → 0.1.0a15__tar.gz

{azureml_registry_tools-0.1.0a13/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a13
+Version: 0.1.0a15
 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<=1.16.86
+Requires-Dist: azureml-assets<2.0
 Dynamic: author
 Dynamic: license
 Dynamic: license-file

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/azureml/registry/_cli/registry_syndication_cli.py

@@ -15,7 +15,6 @@ from azureml.registry.mgmt.create_manifest import generate_syndication_manifest
 from azureml.registry.mgmt.model_management import (
     model_delete, model_get, model_list, parse_asset_id
 )
-from azureml.registry.mgmt.registry_config import create_registry_config
 from azureml.registry.mgmt.syndication_manifest import ResyncAssetsInManifestDto, SyndicationManifest
 
 
@@ -141,14 +140,6 @@ def show_command(registry_name: str, as_arm_object: bool) -> object:
     return discovery
 
 
-def validate_registry_cfg(cfg_file_name) -> str:
-    """Validate if registry config extension is .cfg."""
-    cfg_file_path = Path(cfg_file_name)
-    if not cfg_file_path.suffix == ".cfg":
-        raise argparse.ArgumentTypeError(f"--config arg {cfg_file_name} must be a path to a .cfg file")
-    return cfg_file_path
-
-
 def _add_common_args(p, arg_dicts=None):
     if arg_dicts is None:
         arg_dicts = []
@@ -187,14 +178,11 @@ def main() -> None:
       # Validate an asset
       registry-mgmt asset validate --asset-path ./path-to-asset
 
-      # Deploy an asset to registry (using config file)
-      registry-mgmt asset deploy --asset-path ./path-to-asset --config ./registry-mgmt.cfg
+      # Preview an asset to registry
+      registry-mgmt asset preview --asset-path ./path-to-asset --registry-name myreg --subscription sub123 --resource-group rg123
 
-      # Create registry configuration file
-      registry-mgmt asset config --registry-name myreg --subscription sub123 --resource-group rg123 --tenant-id tenant123
-
-      # Create registry configuration file with storage overrides
-      registry-mgmt asset config --registry-name myreg --subscription sub123 --resource-group rg123 --tenant-id tenant123 --storage-name storage123 --container-name container123 --container-path path123
+      # Deploy an asset to registry
+      registry-mgmt asset deploy --asset-path ./path-to-asset --registry-name myreg --subscription sub123 --resource-group rg123
 
       # Create asset template files
       registry-mgmt asset template --folder ./folder-path
@@ -352,21 +340,19 @@ def main() -> None:
     asset_validate_parser.add_argument("--asset-path", type=Path, help="Path to the asset folder to validate.")
     asset_validate_parser.add_argument("--allow-additional-properties", action="store_true",
                                        help="Allow additional properties not defined in the schema")
+    # asset preview command
+    asset_preview_parser = asset_subparsers.add_parser("preview", help="Preview a model card to registry.")
+    _add_common_args(asset_preview_parser, [registry_name_arg, subscription_arg, resource_group_arg, dry_run_arg])
+    asset_preview_parser.add_argument("--asset-path", type=Path, help="Path to the asset folder to preview.")
+    asset_preview_parser.add_argument("--allow-additional-properties", action="store_true",
+                                      help="Allow additional properties not defined in the schema")
 
     # asset deploy command
-    asset_deploy_parser = asset_subparsers.add_parser("deploy", help="Deploy an asset to registry.")
-    _add_common_args(asset_deploy_parser, [dry_run_arg])
-    asset_deploy_parser.add_argument("-c", "--config", type=validate_registry_cfg, required=True, help="Path to registry config file.")
+    asset_deploy_parser = asset_subparsers.add_parser("deploy", help="Deploy a model to registry.")
+    _add_common_args(asset_deploy_parser, [registry_name_arg, subscription_arg, resource_group_arg, dry_run_arg])
     asset_deploy_parser.add_argument("--asset-path", type=Path, help="Path to the asset folder to deploy.")
-
-    # asset config command
-    asset_config_parser = asset_subparsers.add_parser("config", help="Create registry configuration file.")
-    _add_common_args(asset_config_parser, [registry_name_arg, subscription_arg, resource_group_arg])
-    asset_config_parser.add_argument("--tenant-id", type=str, required=True, help="Registry Tenant ID.")
-    asset_config_parser.add_argument("-c", "--config-file", type=validate_registry_cfg, help="Registry config file path to write to (default: registry-mgmt.cfg).")
-    asset_config_parser.add_argument("--storage-name", type=str, help="Storage account name for storage overrides.")
-    asset_config_parser.add_argument("--container-name", type=str, help="Container name for storage overrides.")
-    asset_config_parser.add_argument("--container-path", type=str, help="Container path for storage overrides.")
+    asset_deploy_parser.add_argument("--allow-additional-properties", action="store_true",
+                                     help="Allow additional properties not defined in the schema")
 
     # asset template command
     asset_template_parser = asset_subparsers.add_parser("template", help="Create asset template files.")
@@ -448,27 +434,23 @@ def main() -> None:
                 syndication_target_set(args.registry_name, args.value, args.dry_run)
     elif args.command == "asset":
         if args.asset_subcommand == "validate":
-            # Config file is not needed for validation
             asset_validate(args.asset_path, args.dry_run, args.allow_additional_properties)
+        elif args.asset_subcommand == "preview":
+            asset_deploy(asset_path=args.asset_path,
+                         registry_name=args.registry_name,
+                         subscription_id=args.subscription,
+                         resource_group=args.resource_group,
+                         dry_run=args.dry_run,
+                         allow_additional_properties=args.allow_additional_properties,
+                         override_storage=True)
         elif args.asset_subcommand == "deploy":
-            # Config file is required for deployment
-            asset_deploy(args.asset_path, args.config, args.dry_run)
-        elif args.asset_subcommand == "config":
-            # Validate storage parameters - all or none must be provided
-            storage_args = [args.storage_name, args.container_name, args.container_path]
-            if any(storage_args) and not all(storage_args):
-                parser.error("When using storage overrides, --storage-name, --container-name, and --container-path are all required.")
-
-            create_registry_config(
-                registry_name=args.registry_name,
-                subscription_id=args.subscription,
-                resource_group=args.resource_group,
-                tenant_id=args.tenant_id,
-                config_file_path=args.config_file,
-                storage_name=args.storage_name,
-                container_name=args.container_name,
-                container_path=args.container_path
-            )
+            asset_deploy(asset_path=args.asset_path,
+                         registry_name=args.registry_name,
+                         subscription_id=args.subscription,
+                         resource_group=args.resource_group,
+                         dry_run=args.dry_run,
+                         allow_additional_properties=args.allow_additional_properties,
+                         override_storage=False)
         elif args.asset_subcommand == "template":
             # Create asset template files
             asset_template(args.folder, args.dry_run)
@@ -516,8 +498,7 @@ def main() -> None:
             if not registry_name or not model_name or not version:
                 print("Error: Registry name, model name, and version must be provided", file=sys.stderr)
                 sys.exit(1)
-
-            # Single call to model_get with args values directly
+            # Get model
             print(model_get(
                 registry_name=registry_name,
                 name=model_name,
@@ -553,7 +534,7 @@ def main() -> None:
                 print("Error: Registry name and model name must be provided", file=sys.stderr)
                 sys.exit(1)
 
-            # Single call to model_delete with args values directly
+            # Delete model
             model_delete(
                 registry_name=registry_name,
                 name=model_name,

azureml_registry_tools-0.1.0a15/azureml/registry/data/description.md.template

@@ -0,0 +1,2 @@
+<!-- REMOVE THESE HEADERS AFTER UPDATING -->
+<!-- `description.md` is required. -->

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/azureml/registry/data/evaluation.md.template

@@ -1,3 +1,2 @@
-<!-- DO NOT CHANGE MARKDOWN HEADERS. IF CHANGED, MODEL CARD MAY BE REJECTED BY A REVIEWER -->
- 
+<!-- REMOVE THESE HEADERS AFTER UPDATING -->
 <!-- `evaluation.md` is highly recommended, but not required. It captures information about the performance of the model. We highly recommend including this section as this information is often used to decide what model to use. -->

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/azureml/registry/data/notes.md.template

@@ -1,3 +1,2 @@
-<!-- DO NOT CHANGE MARKDOWN HEADERS. IF CHANGED, MODEL CARD MAY BE REJECTED BY A REVIEWER -->
- 
-<!-- `notes.md` is highly recommended, but not required. It captures information about how your model is created. We highly recommend including this section to provide transparency for the customers. -->
+<!-- REMOVE THESE HEADERS AFTER UPDATING -->
+<!-- `notes.md` is highly recommended, but not required. It captures information about how your model is created. We highly recommend including this section to provide transparency for the customers. -->

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/azureml/registry/mgmt/asset_management.py

@@ -7,7 +7,7 @@ import sys
 import shutil
 import tempfile
 import yaml
-from datetime import datetime, timedelta
+from datetime import datetime
 from pathlib import Path
 from typing import List
 
@@ -16,7 +16,6 @@ from azure.identity import DefaultAzureCredential
 
 from azureml.registry.data.validate_model_schema import validate_model_schema
 from azureml.registry.data.validate_model_variant_schema import validate_model_variant_schema
-from azureml.registry.mgmt.registry_config import RegistryConfig
 
 # Windows compatibility patch - must be applied before importing azureml.assets
 from subprocess import run
@@ -34,7 +33,7 @@ publish_utils.run_command = patched_run_command
 
 import azureml.assets as assets  # noqa: E402
 import azureml.assets.util as util  # noqa: E402
-from azureml.assets.config import AssetConfig, AssetType, AzureBlobstoreAssetPath  # noqa: E402
+from azureml.assets.config import AssetConfig, AssetType  # noqa: E402
 from azureml.assets.publish_utils import create_asset  # noqa: E402
 from azureml.assets.validate_assets import validate_assets  # noqa: E402
 
@@ -88,34 +87,13 @@ def validate_model(asset_path: Path, allow_additional_properties: bool = False)
     return True
 
 
-def set_storage_and_sas(asset: AssetConfig, storage_config: dict):
-    """Use storage configuration and generate/set SAS token.
-
-    Args:
-        asset (AssetConfig): Asset configuration object to modify
-        storage_config (dict): Storage configuration dictionary
-    """
-    if not storage_config:
-        # No storage overrides provided, skip storage configuration
-        return
-
-    print("Overriding storage configuration with provided values...")
-    extra_config = asset.extra_config_as_object()
-    extra_config._path = AzureBlobstoreAssetPath(
-        storage_name=storage_config["storage_name"],
-        container_name=storage_config["container_name"],
-        container_path=storage_config["container_path"]
-    )
-    _ = extra_config.path.get_uri(token_expiration=timedelta(hours=1))
-
-
-def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, storage_overrides_exist: bool = False) -> AssetConfig:
+def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, override_storage: 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.
+        override_storage (bool, optional): If True, model config will be modified to reference a local temp file.
 
     Returns:
         AssetConfig: Mutable asset configuration object
@@ -136,63 +114,75 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, storage
     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:
-        spec_config = yaml.safe_load(f)
-        spec_config["version"] = datetime.now().strftime("%Y%m%d%H%M%S")
-
-    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"
-
+    # Override storage info for model card preview
+    if override_storage:
+        print("Model card preview - Default setting version and storage info")
+
+        # Autoincrement version for mutable asset
+        with open(spec_config_file, "r") as f:
+            spec_config = yaml.safe_load(f)
+            spec_config["version"] = datetime.now().strftime("%Y%m%d%H%M%S")
+
+        with open(spec_config_file, "w") as f:
+            yaml.dump(spec_config, f)
+
+        # Create dummy file to upload to storage
+        with open(mutable_asset_dir / "dummy.txt", "w") as f:
+            f.write("This is a dummy file used in the artifact for model card preview.")
+
+        yaml_content = """
+        path:
+            uri: ./dummy.txt
+            type: local
+        publish:
+            description: description.md
+            type: custom_model
+        """
+        new_model_config = yaml.safe_load(yaml_content)
+
+        # Overwrite model config
         with open(model_config_file, "w") as f:
-            yaml.dump(model_config, f)
+            yaml.dump(new_model_config, f)
 
     mutable_asset = AssetConfig(asset_config_file)
 
     return mutable_asset
 
 
-def create_or_update_asset(readonly_asset: AssetConfig, config: RegistryConfig):
+def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subscription_id: str, resource_group: str, override_storage: bool = False):
     """Create or update an asset in the AzureML registry.
 
     Args:
         readonly_asset (AssetConfig): Asset configuration to create or update
-        config (RegistryConfig): Registry configuration settings
+        registry_name (str): Name of AzureML registry to deploy to
+        subscription_id (str): Subscription ID of AzureML registry to deploy to
+        resource_group (str): Resource group of AzureML registry to deploy to
+        override_storage (bool): Whether to override storage settings
     """
     print("[CREATING/UPDATING ASSET]")
-    print(f"Using registry configuration from: {config.config_path}")
+    print(f"Registry name: {registry_name}, Subscription ID: {subscription_id}, Resource group: {resource_group}")
+
     # Create ML client
     ml_client = MLClient(
-        subscription_id=config.subscription_id,
-        resource_group_name=config.resource_group,
-        registry_name=config.registry_name,
+        subscription_id=subscription_id,
+        resource_group_name=resource_group,
+        registry_name=registry_name,
         credential=DefaultAzureCredential(),
     )
 
     with tempfile.TemporaryDirectory() as 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
+        mutable_asset = build_mutable_asset(base_asset=readonly_asset, mutable_asset_dir=mutable_asset_dir, override_storage=override_storage)
         try:
-            set_storage_and_sas(mutable_asset, config.storage_config)
-            success = create_asset(mutable_asset, config.registry_name, ml_client)
+            success = create_asset(mutable_asset, registry_name, ml_client)
+            if not success:
+                raise RuntimeError(f"Failed to create/update asset: create_asset 'success' returned {success}")
         except Exception as e:
             print(f"Failed to create/update asset: {e}")
             raise
 
-        if not success:
-            print(f"Failed to create/update asset: create_asset 'success' returned {success}")
-            raise
-
         print("\n[VALIDATE YOUR ASSET IN THE UI HERE]")
-        print(f" - Model Catalog link: https://ai.azure.com/explore/models/{mutable_asset.name}/version/{mutable_asset.version}/registry/{config.registry_name}?tid={config.tenant_id}")
-        print(f" - Azure Portal link: https://ml.azure.com/registries/{config.registry_name}/models/{mutable_asset.name}/version/{mutable_asset.version}?tid={config.tenant_id}")
+        print(f" - Model Catalog link: https://ai.azure.com/explore/models/{mutable_asset.name}/version/{mutable_asset.version}/registry/{registry_name}")
+        print(f" - Azure Portal link: https://ml.azure.com/registries/{registry_name}/models/{mutable_asset.name}/version/{mutable_asset.version}")
 
 
 def asset_validate(asset_path: Path, dry_run: bool = False, allow_additional_properties: bool = False) -> bool:
@@ -237,31 +227,29 @@ def asset_validate(asset_path: Path, dry_run: bool = False, allow_additional_pro
     return validate_model(readonly_asset.file_path, allow_additional_properties)
 
 
-def asset_deploy(asset_path: Path, config_path: Path, dry_run: bool = False) -> bool:
-    """Deploy an asset using configuration file.
+def asset_deploy(asset_path: Path, registry_name: str, subscription_id: str, resource_group: str, dry_run: bool = False, allow_additional_properties: bool = False, override_storage: bool = False) -> bool:
+    """Deploy an asset to a registry.
 
     Args:
         asset_path (Path): Path to the asset folder to deploy
-        config_path (Path): Path to configuration file
         dry_run (bool): If True, perform a dry run without deploying
+        allow_additional_properties (bool): Whether to allow additional properties not defined in schema
+        override_storage (bool): Whether to override storage settings
 
     Returns:
         bool: True if deployment succeeds, False otherwise
     """
-    try:
-        config = RegistryConfig(config_path)
-    except Exception as e:
-        print(f"❌ [ERROR]: Configuration validation failed: {e}")
-        return False
-
     if dry_run:
-        print(f"[DRY RUN] Would deploy asset at {asset_path} to registry {config.registry_name}")
+        if override_storage:
+            print(f"[DRY RUN] Would preview asset at {asset_path} to registry {registry_name}")
+        else:
+            print(f"[DRY RUN] Would deploy asset at {asset_path} to registry {registry_name}")
         return True
 
     asset_path = asset_path.resolve()
 
     # Validate asset before deployment
-    if not asset_validate(asset_path, dry_run=False):
+    if not asset_validate(asset_path, dry_run=False, allow_additional_properties=allow_additional_properties):
         print("❌ [ERROR]: Asset validation failed. Asset deployment aborted.")
         return False
 
@@ -269,7 +257,7 @@ def asset_deploy(asset_path: Path, config_path: Path, dry_run: bool = False) ->
     readonly_asset = assets.AssetConfig(asset_path / assets.DEFAULT_ASSET_FILENAME)
 
     try:
-        create_or_update_asset(readonly_asset, config)
+        create_or_update_asset(readonly_asset, registry_name, subscription_id, resource_group, override_storage)
         return True
     except Exception as e:
         print(f"❌ [ERROR]: Failed to deploy asset: {e}")

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15/azureml_registry_tools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a13
+Version: 0.1.0a15
 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<=1.16.86
+Requires-Dist: azureml-assets<2.0
 Dynamic: author
 Dynamic: license
 Dynamic: license-file

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/azureml_registry_tools.egg-info/SOURCES.txt

@@ -28,7 +28,6 @@ azureml/registry/mgmt/create_asset_template.py
 azureml/registry/mgmt/create_manifest.py
 azureml/registry/mgmt/create_model_spec.py
 azureml/registry/mgmt/model_management.py
-azureml/registry/mgmt/registry_config.py
 azureml/registry/mgmt/syndication_manifest.py
 azureml/registry/tools/__init__.py
 azureml/registry/tools/config.py

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/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<=1.16.86
+azureml-assets<2.0

{azureml_registry_tools-0.1.0a13 → azureml_registry_tools-0.1.0a15}/setup.py

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

azureml_registry_tools-0.1.0a13/azureml/registry/data/description.md.template

@@ -1,3 +0,0 @@
-<!-- DO NOT CHANGE MARKDOWN HEADERS. IF CHANGED, MODEL CARD MAY BE REJECTED BY A REVIEWER -->
- 
-<!-- `description.md` is required. -->

azureml_registry_tools-0.1.0a13/azureml/registry/mgmt/registry_config.py

@@ -1,147 +0,0 @@
-# ---------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
-# ---------------------------------------------------------
-
-"""RegistryConfig class."""
-
-import configparser
-from pathlib import Path
-from typing import Dict
-
-
-class RegistryConfig:
-    """RegistryConfig class."""
-
-    def __init__(self, config_path: Path = None):
-        """Registry config init.
-
-        Args:
-            config_path (Path, optional): Path to configuration file. Defaults to "registry-mgmt.cfg".
-        """
-        if config_path is None:
-            print("No config path provided, using default 'registry-mgmt.cfg'.")
-            config_path = Path("registry-mgmt.cfg")
-        self.config_path = config_path
-        if not self.config_path.is_file():
-            raise FileNotFoundError(f"File '{self.config_path.resolve().as_posix()}' does not exist.")
-
-        self.config = configparser.ConfigParser()
-        self.config.read(self.config_path)
-        self._validate_schema()
-
-    def _validate_schema(self) -> None:
-        """Validate registry config schema."""
-        config_file_name = str(self.config_path)
-
-        if not self.config.has_section("registry"):
-            raise Exception(f'"registry" section not found in config file {config_file_name}')
-
-        # Validate registry section
-        if not self.config.has_option("registry", "registry_name"):
-            raise Exception(f'Key "registry_name" not found under "registry" section in config file {config_file_name}')
-        if not self.config.has_option("registry", "tenant_id"):
-            raise Exception(f'Key "tenant_id" not found under "registry" section in config file {config_file_name}')
-        if not self.config.has_option("registry", "subscription_id"):
-            raise Exception(f'Key "subscription_id" not found under "registry" section in config file {config_file_name}')
-        if not self.config.has_option("registry", "resource_group"):
-            raise Exception(f'Key "resource_group" not found under "registry" section in config file {config_file_name}')
-
-        # Validate that values are not empty
-        required_settings = ["registry_name", "tenant_id", "subscription_id", "resource_group"]
-        for setting in required_settings:
-            if not self.config.get("registry", setting).strip():
-                raise Exception(f'Value for "{setting}" is empty in "registry" section in config file {config_file_name}')
-
-        # Validate storage_overrides section (optional, but if present, validate its contents)
-        if self.config.has_section("storage_overrides"):
-            allowed_storage_keys = {"storage_name", "container_name", "container_path"}
-            actual_storage_keys = set(self.config.options("storage_overrides"))
-
-            # Check for unexpected keys
-            unexpected_keys = actual_storage_keys - allowed_storage_keys
-            if unexpected_keys:
-                raise Exception(f'Unexpected keys {sorted(unexpected_keys)} found in "storage_overrides" section in config file {config_file_name}. Only {sorted(allowed_storage_keys)} are allowed.')
-
-            # Check for required keys
-            if not self.config.has_option("storage_overrides", "storage_name"):
-                raise Exception(f'Key "storage_name" not found under "storage_overrides" section in config file {config_file_name}')
-            if not self.config.has_option("storage_overrides", "container_name"):
-                raise Exception(f'Key "container_name" not found under "storage_overrides" section in config file {config_file_name}')
-            if not self.config.has_option("storage_overrides", "container_path"):
-                raise Exception(f'Key "container_path" not found under "storage_overrides" section in config file {config_file_name}')
-
-            # Validate that storage values are not empty
-            required_storage_settings = ["storage_name", "container_name", "container_path"]
-            for setting in required_storage_settings:
-                if not self.config.get("storage_overrides", setting).strip():
-                    raise Exception(f'Value for "{setting}" is empty in "storage_overrides" section in config file {config_file_name}')
-
-    @property
-    def registry_name(self) -> str:
-        """Get registry name from config."""
-        return self.config.get("registry", "registry_name")
-
-    @property
-    def tenant_id(self) -> str:
-        """Get tenant ID from config."""
-        return self.config.get("registry", "tenant_id")
-
-    @property
-    def subscription_id(self) -> str:
-        """Get subscription ID from config."""
-        return self.config.get("registry", "subscription_id")
-
-    @property
-    def resource_group(self) -> str:
-        """Get resource group from config."""
-        return self.config.get("registry", "resource_group")
-
-    @property
-    def storage_config(self) -> Dict[str, str]:
-        """Get storage configuration."""
-        if self.config.has_section("storage_overrides"):
-            return dict(self.config["storage_overrides"])
-        return {}
-
-
-def create_registry_config(registry_name: str, subscription_id: str, resource_group: str, tenant_id: str,
-                           config_file_path: Path, storage_name: str = None, container_name: str = None,
-                           container_path: str = None):
-    """Create registry config.
-
-    Args:
-        registry_name (str): Registry name.
-        subscription_id (str): Registry subscription id.
-        resource_group (str): Registry resource group.
-        tenant_id (str): Tenant ID.
-        config_file_path (Path): Path to config file.
-        storage_name (str, optional): Storage account name for storage overrides. Defaults to None.
-        container_name (str, optional): Container name for storage overrides. Defaults to None.
-        container_path (str, optional): Container path for storage overrides. Defaults to None.
-    """
-    print("Creating registry config...")
-
-    registry_config = configparser.ConfigParser()
-    registry_config.add_section("registry")
-    registry_config.set("registry", "registry_name", registry_name)
-    registry_config.set("registry", "subscription_id", subscription_id)
-    registry_config.set("registry", "resource_group", resource_group)
-    registry_config.set("registry", "tenant_id", tenant_id)
-
-    # Add storage overrides if provided
-    if storage_name and container_name and container_path:
-        registry_config.add_section("storage_overrides")
-        registry_config.set("storage_overrides", "storage_name", storage_name)
-        registry_config.set("storage_overrides", "container_name", container_name)
-        registry_config.set("storage_overrides", "container_path", container_path)
-
-    # Write to path
-    if config_file_path is None:
-        print('No config file path provided, using default "registry-mgmt.cfg"')
-        config_file_path = Path("registry-mgmt.cfg")
-    with open(config_file_path, "w") as config_file:
-        registry_config.write(config_file)
-
-    config_abs_path = Path(config_file_path).resolve().as_posix()
-
-    print(f"Wrote registry config file to {config_abs_path}")