azureml-registry-tools 0.1.0a30__tar.gz → 0.1.0a32__tar.gz

{azureml_registry_tools-0.1.0a30/azureml_registry_tools.egg-info → azureml_registry_tools-0.1.0a32}/PKG-INFO

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

{azureml_registry_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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|reasoning-summary|streaming|tool-calling|function-calling|image-input)(?:\\s*,\\s*(agents|agentsV2|assistants|routing|reasoning|reasoning-summary|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",
@@ -495,6 +495,7 @@
         "invisibleUntil": {
           "type": "string",
           "format": "date-time",
+          "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
           "description": "Date until which the model is invisible"
         },
         "playgroundRateLimitTier": {
@@ -605,16 +606,6 @@
           },
           "description": "Model capabilities (e.g., agents, assistants)"
         },
-        "playground_rate_limit_tier": {
-          "type": "string",
-          "description": "Rate limit tier for playground",
-          "enum": [
-            "high",
-            "low",
-            "custom",
-            "embeddings"
-          ]
-        },
         "azureOffers": {
           "type": "array",
           "items": {
@@ -643,7 +634,8 @@
         },
         "sharedComputeCapacityEnabled": {
           "type": "boolean",
-          "description": "Whether shared compute capacity is enabled"
+          "enum": [true],
+          "description": "Whether shared compute capacity is enabled. Only true values are allowed - remove this field when disabled."
         },
         "freePlayground": {
           "type": "boolean",
@@ -678,7 +670,8 @@
         },
         "isDirectFromAzure": {
           "type": "boolean",
-          "description": "If the model is an Azure Direct model"
+          "enum": [true],
+          "description": "If the model is an Azure Direct model. Only true values are allowed - remove this field when disabled."
         },
         "inferenceEnvironmentAssetId": {
           "type": "string",
@@ -855,14 +848,17 @@
             }
           }
         },
-        "sku": {
+        "SKU": {
           "type": "object",
           "description": "SKU specifications for different operations",
           "additionalProperties": false,
           "properties": {
-            "finetuneRecommendedSku": {
-              "description": "Skus that can be used for finetune maap",
-              "$ref": "#/definitions/skuListDefinition"
+            "finetuneRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "Skus that can be used for finetune maap"
             },
             "evaluationMinSkuSpec": {
               "type": "string",
@@ -870,9 +866,12 @@
               "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
               "format": "sku-spec"
             },
-            "evaluationRecommendedSku": {
-              "description": "Evaluation recommended sku",
-              "$ref": "#/definitions/skuListDefinition"
+            "evaluationRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "Evaluation recommended skus"
             },
             "finetuneMinSkuSpec": {
               "type": "string",
@@ -886,9 +885,12 @@
               "pattern": "^\\d+\\|\\d+\\|\\d+\\|\\d+$",
               "format": "sku-spec"
             },
-            "inferenceRecommendedSku": {
-              "description": "A list of comma separated skus that the client should use to deploy the model",
-              "$ref": "#/definitions/skuListDefinition"
+            "inferenceRecommendedSkus": {
+              "type": "array",
+              "items": {
+                "$ref": "#/definitions/skuEnum"
+              },
+              "description": "A list of skus that the client should use to deploy the model"
             }
           }
         },
@@ -899,7 +901,8 @@
           "properties": {
             "maasFinetuningTest": {
               "type": "boolean",
-              "description": "Used for testing maas-finetuning in INT"
+              "enum": [true],
+              "description": "Used for testing maas-finetuning in INT. Only true values are allowed - remove this field when disabled."
             },
             "maasFinetuningDeployRegions": {
               "type": "array",
@@ -918,12 +921,23 @@
             },
             "maasFinetuningOffer": {
               "type": "boolean",
-              "description": "Whether MaaS fine-tuning offer is available"
+              "enum": [true],
+              "description": "Whether MaaS fine-tuning offer is available. Only true values are allowed - remove this field when disabled."
             },
-            "maasFinetuningRegions": {
-              "type": "string",
-              "description": "Regions where fine-tuning is available as a comma-separated string",
-              "pattern": "^(westus3|eastus2|eastus|southcentralus|northcentralus|westus)(?:\\s*,\\s*(westus3|eastus2|eastus|southcentralus|northcentralus|westus))*$"
+            "maasFinetuningAzureRegions": {
+              "type": "array",
+              "items": {
+                "type": "string",
+                "enum": [
+                  "westus3",
+                  "eastus2",
+                  "eastus",
+                  "southcentralus",
+                  "northcentralus",
+                  "westus"
+                ]
+              },
+              "description": "Regions where fine-tuning is available"
             }
           }
         },
@@ -934,32 +948,32 @@
           "properties": {
             "finetuneDeprecationDate": {
               "type": "string",
-              "description": "The date that finetune will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that finetune will be deprecated. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "finetuneRetirementDate": {
               "type": "string",
-              "description": "The date that finetune will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that finetune will be retired. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceDeprecationDate": {
               "type": "string",
-              "description": "The date that inference will be deprecated. Date format yyyy-MM-dd. The deprecation will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will be deprecated. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceLegacyDate": {
               "type": "string",
-              "description": "The date that inference will become legacy. Date format yyyy-MM-dd. The change will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will become legacy. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             },
             "inferenceRetirementDate": {
               "type": "string",
-              "description": "The date that inference will be retired. Date format yyyy-MM-dd. The retirement will happen at midnight UTC",
-              "pattern": "^\\d{4}-\\d{2}-\\d{2}$",
+              "description": "The date that inference will be retired. This date-time should be in midnight UTC.",
+              "pattern": "^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}\\.\\d{1,6}Z$",
               "format": "date-time"
             }
           }

{azureml_registry_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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)
 
@@ -236,7 +236,7 @@ def create_or_update_asset(readonly_asset: AssetConfig, registry_name: str, subs
         subscription_id=subscription_id,
         resource_group_name=resource_group,
         registry_name=registry_name,
-        credential=DefaultAzureCredential(),
+        credential=DefaultAzureCredential(),  # CodeQL [SM05139] DefaultAzureCredential should only be used for local development and testing purposes.
     )
 
     with tempfile.TemporaryDirectory() as mutable_asset_dir:

{azureml_registry_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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-0.1.0a30 → azureml_registry_tools-0.1.0a32}/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.0a30 → azureml_registry_tools-0.1.0a32/azureml_registry_tools.egg-info}/PKG-INFO

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

{azureml_registry_tools-0.1.0a30 → azureml_registry_tools-0.1.0a32}/setup.py

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