azureml-registry-tools 0.1.0a29__py3-none-any.whl → 0.1.0a31__py3-none-any.whl

azureml/registry/data/model.schema.json

@@ -304,10 +304,10 @@
           "type": "string",
           "format": "long-string"
         },
-        "fineTuningMethods": {
-          "description": "Fine-tuning strategy applied to customize the model behavior",
+        "fineTuneOnlyModel": {
+          "description": "Indicates the model is available only for fine-tuning. The model itself is not deployable for inferencing.",
           "type": "string",
-          "pattern": "^(Supervised|Direct Preference Optimization|Reinforcement)(?:\\s*,\\s*(Supervised|Direct Preference Optimization|Reinforcement))*$"
+          "enum": ["true", "false"]
         },
         "maas-finetuning": {
           "description": "Whether MaaS fine-tuning is enabled. True means enabled, if disabled remove the key",
@@ -361,7 +361,7 @@
         "modelCapabilities": {
           "description": "Model capabilities (e.g., agents, assistants) as a comma-separated string",
           "type": "string",
-          "pattern": "^(agents|agentsV2|assistants|routing|reasoning|streaming|tool-calling|function-calling|image-input)(?:\\s*,\\s*(agents|agentsV2|assistants|routing|reasoning|streaming|tool-calling|function-calling|image-input))*$"
+          "pattern": "^(agents|agentsV2|assistants|routing|reasoning|reasoning-summary|streaming|tool-calling|function-calling|image-input|fine-tuning)(?:\\s*,\\s*(agents|agentsV2|assistants|routing|reasoning|reasoning-summary|streaming|tool-calling|function-calling|image-input|fine-tuning))*$"
         },
         "modelHash": {
           "description": "Hash of the model",

azureml/registry/data/validate_model_schema.py

@@ -40,7 +40,7 @@ def load_schema(schema_file: Path, allow_additional_properties: bool = False) ->
         dict: Loaded model schema
     """
     # Load schema from file
-    with open(schema_file, 'r') as file:
+    with open(schema_file, 'r', encoding='utf-8') as file:
         schema = yaml.safe_load(file)
 
     if allow_additional_properties:
@@ -101,7 +101,7 @@ def validate_model_schema(input_dirs: List[Path],
                     changed_model_count += 1
                     # Validate the file against the schema
                     try:
-                        with open(file_path, "r") as f:
+                        with open(file_path, "r", encoding="utf-8") as f:
                             spec_config = yaml.safe_load(f)
 
                         # Collect all validation errors
@@ -123,7 +123,7 @@ def validate_model_schema(input_dirs: List[Path],
                                 else:
                                     # Try to find line number by looking at the path and instance
                                     try:
-                                        with open(file_path, "r") as f:
+                                        with open(file_path, "r", encoding="utf-8") as f:
                                             yaml_content = f.readlines()
                                             yaml_lines = []
                                             for idx, line in enumerate(yaml_content):

azureml/registry/data/validate_model_variant_schema.py

@@ -32,7 +32,7 @@ def validate_model_variant_schema(input_dirs: List[Path],
     """
     # Load variantInfo schema from file
     model_variant_info_schema = {}
-    with open(model_variant_schema_file, 'r') as file:
+    with open(model_variant_schema_file, 'r', encoding="utf-8") as file:
         model_variant_info_schema = yaml.safe_load(file)
 
     asset_count = 0
@@ -45,7 +45,7 @@ def validate_model_variant_schema(input_dirs: List[Path],
                 model_count += 1
                 # Extract model variant info from spec
                 variant_info = None
-                with open(asset_config.spec_with_path, "r") as f:
+                with open(asset_config.spec_with_path, "r", encoding="utf-8") as f:
                     spec_config = yaml.safe_load(f)
                     variant_info = spec_config.get("variantInfo")
 

azureml/registry/mgmt/asset_management.py

@@ -167,7 +167,7 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
     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)
 
-    with open(spec_config_file, "r") as f:
+    with open(spec_config_file, "r", encoding="utf-8") as f:
         spec_config = yaml.safe_load(f)
 
     # Override storage info for model card preview
@@ -182,7 +182,7 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
             spec_config["properties"].pop("intellectualPropertyPublisher", None)
 
         # Create dummy file to upload to storage
-        with open(mutable_asset_dir / "dummy.txt", "w") as f:
+        with open(mutable_asset_dir / "dummy.txt", "w", encoding="utf-8") as f:
             f.write("This is a dummy file used in the artifact for model card preview.")
 
         yaml_content = """
@@ -196,8 +196,8 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
         new_model_config = yaml.safe_load(yaml_content)
 
         # Overwrite model config
-        with open(model_config_file, "w") as f:
-            yaml.dump(new_model_config, f)
+        with open(model_config_file, "w", encoding="utf-8") as f:
+            yaml.dump(new_model_config, f, allow_unicode=True, default_flow_style=False)
 
     # Extract system_metadata from model spec
     # PUT request will be made after asset creation while the SDK update/PATCH logic is being updated
@@ -210,8 +210,8 @@ def build_mutable_asset(base_asset: AssetConfig, mutable_asset_dir: str, overrid
 
     # Write updated spec config back to the file
     if override_storage or system_metadata_payload:
-        with open(spec_config_file, "w") as f:
-            yaml.dump(spec_config, f)
+        with open(spec_config_file, "w", encoding="utf-8") as f:
+            yaml.dump(spec_config, f, allow_unicode=True, default_flow_style=False)
 
     mutable_asset = AssetConfig(asset_config_file)
 

azureml/registry/mgmt/util.py

@@ -66,7 +66,7 @@ def _resolve_from_file(value: Union[str, Path]) -> Tuple[bool, Union[str, None]]
     """
     if os.path.isfile(value):
         try:
-            with open(value, 'r') as f:
+            with open(value, 'r', encoding='utf-8') as f:
                 content = f.read()
                 return (True, content)
         except Exception as e:

azureml/registry/tools/config.py

@@ -33,7 +33,7 @@ class AssetSpec:
 
     def __init__(self, file_name: str, asset_type: str):
         """Asset spec init."""
-        with open(file_name) as f:
+        with open(file_name, "r", encoding="utf-8") as f:
             self._yaml = YAML().load(f)
 
         self._file_name_with_path = Path(file_name)

azureml/registry/tools/create_or_update_assets.py

@@ -132,7 +132,7 @@ def merge_yamls(existing_asset_file_name: str,
     Returns:
         dict: Merged asset data.
     """
-    with open(existing_asset_file_name, "r") as existing_asset_file, open(updated_asset.file_name_with_path, "r") as updated_asset_file:
+    with open(existing_asset_file_name, "r", encoding="utf-8") as existing_asset_file, open(updated_asset.file_name_with_path, "r", encoding="utf-8") as updated_asset_file:
         existing_asset_dict = yaml.safe_load(existing_asset_file)
         updated_asset_dict = yaml.safe_load(updated_asset_file)
 
@@ -173,8 +173,8 @@ def merge_assets(existing_asset: Union[Component, Data, Model],
     with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as merged_asset_temp_file:
         merged_asset_temp_file_name = merged_asset_temp_file.name
 
-    with open(merged_asset_temp_file_name, "w") as merged_asset_temp_file:
-        merged_asset_temp_file.write(yaml.dump(merged_result))
+    with open(merged_asset_temp_file_name, "w", encoding="utf-8") as merged_asset_temp_file:
+        merged_asset_temp_file.write(yaml.dump(merged_result, allow_unicode=True))
 
     if asset.type == AssetType.MODEL:
         merged_asset = load_model(merged_asset_temp_file_name)
@@ -211,8 +211,8 @@ def write_results(results_dict: dict,
     # Produce JSON output on status and diffs
     print("\nAsset create_or_update results:")
     print(json.dumps(results_dict, indent=2))
-    with open(output_json_file_name, "w") as f:
-        json.dump(results_dict, f, indent=2)
+    with open(output_json_file_name, "w", encoding="utf-8") as f:
+        json.dump(results_dict, f, indent=2, ensure_ascii=False)
         print(f"Wrote asset create_or_update results to file {output_json_file_name}")
 
 

azureml/registry/tools/repo2registry_config.py

@@ -100,7 +100,7 @@ def create_repo2registry_config(registry_name: str, subscription: str, resource_
     repo2registry_config.set("settings", "continue_on_asset_failure", continue_on_asset_failure)
 
     # Write to path
-    with open(repo2registry_config_file_name, "w") as repo2registry_config_file:
+    with open(repo2registry_config_file_name, "w", encoding="utf-8") as repo2registry_config_file:
         repo2registry_config.write(repo2registry_config_file)
 
     repo2registry_cfg_abs_path = Path(repo2registry_config_file_name).resolve().as_posix()

{azureml_registry_tools-0.1.0a29.dist-info → azureml_registry_tools-0.1.0a31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: azureml-registry-tools
-Version: 0.1.0a29
+Version: 0.1.0a31
 Summary: AzureML Registry tools and CLI
 Author: Microsoft Corp
 License: https://aka.ms/azureml-sdk-license
@@ -9,7 +9,7 @@ License-File: LICENSE.txt
 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: azure-ai-ml<2.0,>=1.30.0
 Requires-Dist: azureml-assets<2.0
 Dynamic: author
 Dynamic: license

{azureml_registry_tools-0.1.0a29.dist-info → azureml_registry_tools-0.1.0a31.dist-info}/RECORD

@@ -13,27 +13,27 @@ azureml/registry/data/asset.yaml.template,sha256=WTgfuvKEBp-EVFSQ0JpU0h4z_ULJdUL
 azureml/registry/data/description.md.template,sha256=DiVAQEXGXoKmhV4LPqE3NupxVtcsuDZ1pn2UA5Fzd6U,2821
 azureml/registry/data/evaluation.md.template,sha256=FC9U8EI_1Dg9Vz18ftAFDDmTqvpwELDkIDlYqB8C9Dk,1031
 azureml/registry/data/model-variant.schema.json,sha256=AT4Dy6cCtp_SFUfSqYIqcER8AldpYm0QIEy1abY3QWE,1699
-azureml/registry/data/model.schema.json,sha256=GbpT9kqZ-7t_2YxoU4QX1ojI_yo9XJfR2bAhrG0LgZc,47085
+azureml/registry/data/model.schema.json,sha256=BtzhaZFp2FeQMBcTsYLj5KoC9l36Zvzmt_k2v4hti9Y,47073
 azureml/registry/data/model.yaml.template,sha256=h5uqAN22FLaWrbPxIb8yVKH9cGDBrIwooXYYfsKhxDw,245
 azureml/registry/data/notes.md.template,sha256=rgGGHQaxfVg6COIzZU8EVBa48sRPnNezVkCMGNyRRNo,1528
-azureml/registry/data/validate_model_schema.py,sha256=OQp2E01kdxSphvUQYQvelSiD24-qUG6nTFuzW60wX2c,8322
-azureml/registry/data/validate_model_variant_schema.py,sha256=JPVNtRBn6qciMu4PaRXOvS86OGGW0cocL2Rri4xYKo8,3629
+azureml/registry/data/validate_model_schema.py,sha256=4Exd9K6ry93URbx21gn2wSJ6xPacYfO7tmZ96cr8ViM,8376
+azureml/registry/data/validate_model_variant_schema.py,sha256=0nC1zWzvtYAwowY1MG4lIhFDtBBhezwpya4z-ItafWo,3665
 azureml/registry/mgmt/__init__.py,sha256=LMhqcEC8ItmmpKZljElGXH-6olHlT3SLl0dJU01OvuM,226
-azureml/registry/mgmt/asset_management.py,sha256=NeYjjtOFlXJT1c0s9pW5mBjJgyxQ7zqZbB-YzmNOg9s,14465
+azureml/registry/mgmt/asset_management.py,sha256=Gnkea8N2UTfZljkPuQ1ZLmEY8ejFaMYZSB3DjmjXJ0U,14629
 azureml/registry/mgmt/create_asset_template.py,sha256=ejwLuIsmzJOoUePoxbM-eGMg2E3QHfdX-nPMBzYUVMQ,3525
 azureml/registry/mgmt/create_manifest.py,sha256=N9wRmjAKO09A3utN_lCUsM_Ufpj7PL0SJz-XHPHWuyM,9528
 azureml/registry/mgmt/create_model_spec.py,sha256=1PdAcUf-LomvljoT8wKQihXMTLd7DoTgN0qDX4Lol1A,10473
 azureml/registry/mgmt/model_management.py,sha256=STTr_uvdPKV2NaJ5UvS5aMi3yejVF6Hkj9DjofJLQik,7453
 azureml/registry/mgmt/syndication_manifest.py,sha256=8Sfd49QuCA5en5_mIOLE21kZVpnReUXowx_g0TVRgWg,9025
-azureml/registry/mgmt/util.py,sha256=BeVUsiMbZdO6rc9tHlsFg0AkVX62lzKn4Ko_y329r-I,2385
+azureml/registry/mgmt/util.py,sha256=1q90sLr70w8FrXYqyHUwszjXa7OdT2MRSI_gJf-aVdY,2403
 azureml/registry/tools/__init__.py,sha256=IAuWWpGfZm__pAkBIxmpJz84QskpkxBr0yDk1TUSnkE,223
-azureml/registry/tools/config.py,sha256=tjPaoBsWtPXBL8Ww1hcJtsr2SuIjPKt79dR8iovcebg,3639
-azureml/registry/tools/create_or_update_assets.py,sha256=7LcuBzwU-HNE9ADG2igFXI696aKR028yTYxMuDtjVmA,16095
+azureml/registry/tools/config.py,sha256=eZ8F6vKvKX-rHkZrnLmjX-h_1ccqSSTkUK0s3ADoD9o,3662
+azureml/registry/tools/create_or_update_assets.py,sha256=xKGmcc3GVQ0qoZav68NENL86dYN1jdTPXCHK1SfK_vM,16207
 azureml/registry/tools/registry_utils.py,sha256=zgYlCiOONtQJ4yZ9wg8tKVoE8dh6rrjB8hYBGhpV9-0,1403
-azureml/registry/tools/repo2registry_config.py,sha256=eXp_tU8Jyi30g8xGf7wbpLgKEPpieohBANKxMSLzq7s,4873
-azureml_registry_tools-0.1.0a29.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
-azureml_registry_tools-0.1.0a29.dist-info/METADATA,sha256=l6pLVmMr-nw8Cd7XPT2XSlcxd2r6prXEkAK8ch6OGqE,522
-azureml_registry_tools-0.1.0a29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-azureml_registry_tools-0.1.0a29.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
-azureml_registry_tools-0.1.0a29.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
-azureml_registry_tools-0.1.0a29.dist-info/RECORD,,
+azureml/registry/tools/repo2registry_config.py,sha256=YxSpLl16m-Cs9RkrXjpWgg6la2zukl-y1i12pV4UIco,4891
+azureml_registry_tools-0.1.0a31.dist-info/licenses/LICENSE.txt,sha256=n20rxwp7_NGrrShv9Qvcs90sjI1l3Pkt3m-5OPCWzgs,845
+azureml_registry_tools-0.1.0a31.dist-info/METADATA,sha256=xs_wccLvrMlgq5hb8yhSy7HNJ9tHU_6kbAavpRub_HU,531
+azureml_registry_tools-0.1.0a31.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+azureml_registry_tools-0.1.0a31.dist-info/entry_points.txt,sha256=iRUkAeQidMnO6RQzpLqMUBTcyYtNzAfSin9WnSdVGLw,147
+azureml_registry_tools-0.1.0a31.dist-info/top_level.txt,sha256=ZOeEa0TAXo6i5wOjwBoqfIGEuxOcKuscGgNSpizqREY,8
+azureml_registry_tools-0.1.0a31.dist-info/RECORD,,