Skip to content

Commit e329366

Browse files
authored
Some bug fixes on package model (Azure#30336)
* some bug fixes * client change
1 parent 494261b commit e329366

File tree

4 files changed

+6
-3
lines changed

4 files changed

+6
-3
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/assets/package/model_configuration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
class ModelConfigurationSchema(PathAwareSchema):
1818
mode = StringTransformedEnum(
1919
allowed_values=[
20-
"readonly_mount",
20+
"read_only_mount",
2121
"download",
2222
]
2323
)

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/assets/package/model_package_input.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ class ModelPackageInputSchema(PathAwareSchema):
6060
type = StringTransformedEnum(allowed_values=["uri_file", "uri_folder"], casing_transform=camel_to_snake)
6161
mode = StringTransformedEnum(
6262
allowed_values=[
63-
"readonly_mount",
63+
"read_only_mount",
6464
"download",
6565
],
6666
casing_transform=camel_to_snake,

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_assets/_artifacts/_package/model_configuration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from azure.ai.ml._restclient.v2023_04_01_preview.models import ModelConfiguration as RestModelConfiguration
77
from azure.ai.ml._utils._experimental import experimental
8+
from azure.ai.ml._utils.utils import snake_to_camel
89

910

1011
@experimental
@@ -26,4 +27,4 @@ def _from_rest_object(cls, rest_obj: RestModelConfiguration) -> "ModelConfigurat
2627
return ModelConfiguration(mode=rest_obj.mode, mount_path=rest_obj.mount_path)
2728

2829
def _to_rest_object(self) -> RestModelConfiguration:
29-
return RestModelConfiguration(mode=self.mode, mount_path=self.mount_path)
30+
return RestModelConfiguration(mode=snake_to_camel(self.mode), mount_path=self.mount_path)

sdk/ml/azure-ai-ml/azure/ai/ml/operations/_model_operations.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -607,6 +607,8 @@ def package(self, name: str, version: str, package_request: ModelPackage, **kwar
607607

608608
package_request.base_environment_source.resource_id = (
609609
"azureml:/" + package_request.base_environment_source.resource_id
610+
if not package_request.base_environment_source.resource_id.startswith(ARM_ID_PREFIX)
611+
else package_request.base_environment_source.resource_id
610612
)
611613

612614
package_request = package_request._to_rest_object()

0 commit comments

Comments
 (0)