Skip to content

Normalize the managed identity type in pipeline command and parallel node #34435

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
wangchao1230 merged 11 commits into from
Mar 2, 2024
Merged

Normalize the managed identity type in pipeline command and parallel node #34435

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
bf02f22
allow prs run settings binding to literal input
smile21prc Mar 15, 2023
16485b4
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
smile21prc Mar 15, 2023
d5eeb29
fix conflict
bupt-wenxiaole Apr 10, 2023
dd128dc
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
bupt-wenxiaole Oct 17, 2023
1e27e66
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
bupt-wenxiaole Dec 13, 2023
5183e82
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
bupt-wenxiaole Jan 3, 2024
25a03c5
Merge branch 'main' of https://github.com/Azure/azure-sdk-for-python
bupt-wenxiaole Feb 21, 2024
4b4c5a1
fix managed identity issue
bupt-wenxiaole Feb 23, 2024
0786a7a
refine
bupt-wenxiaole Feb 23, 2024
3e5e296
revert
bupt-wenxiaole Feb 23, 2024
b5c2d32
fix UT
bupt-wenxiaole Mar 1, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix managed identity issue
  • Loading branch information
@bupt-wenxiaole
bupt-wenxiaole committed Feb 23, 2024
commit 4b4c5a1a919d7414e9a5688642dcb9947295bdd4
2 changes: 1 addition & 1 deletion sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/command.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -771,7 +771,7 @@ def _to_rest_object(self, **kwargs: Any) -> dict:
"limits": get_rest_dict_for_node_attrs(self.limits, clear_empty_value=True),
"resources": get_rest_dict_for_node_attrs(self.resources, clear_empty_value=True),
"services": get_rest_dict_for_node_attrs(self.services),
"identity": self.identity._to_dict() if self.identity and not isinstance(self.identity, Dict) else None,
"identity": get_rest_dict_for_node_attrs(self.identity),
"queue_settings": get_rest_dict_for_node_attrs(self.queue_settings, clear_empty_value=True),
}.items():
if value is not None:
Expand Down
7 changes: 2 additions & 5 deletions sdk/ml/azure-ai-ml/azure/ai/ml/entities/_builders/parallel.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -450,9 +450,7 @@ def _to_rest_object(self, **kwargs: Any) -> dict:
"partition_keys": json.dumps(self.partition_keys)
if self.partition_keys is not None
else self.partition_keys,
"identity": self.identity._to_dict()
if self.identity and not isinstance(self.identity, Dict)
else None,
"identity": get_rest_dict_for_node_attrs(self.identity),
"resources": get_rest_dict_for_node_attrs(self.resources),
}
)
Expand Down Expand Up @@ -481,9 +479,8 @@ def _from_rest_object_to_init_params(cls, obj: dict) -> Dict:

if "partition_keys" in obj and obj["partition_keys"]:
obj["partition_keys"] = json.dumps(obj["partition_keys"])

if "identity" in obj and obj["identity"]:
obj["identity"] = _BaseJobIdentityConfiguration._load(obj["identity"])
obj["identity"] = _BaseJobIdentityConfiguration._from_rest_object(obj["identity"])
return obj

def _build_inputs(self) -> Dict:
Expand Down