From 0c6fb456352c290bf983de7c9563385c0d45c4ed Mon Sep 17 00:00:00 2001 From: Ayushh Garg Date: Tue, 5 May 2026 13:01:43 +0530 Subject: [PATCH] Tighten regex in artifact tool download --- sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_artifact_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_artifact_utils.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_artifact_utils.py index 9b00040ef19c..970eb3e62e68 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_artifact_utils.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_artifact_utils.py @@ -171,12 +171,12 @@ def _redirect_artifacts_tool_path(self, organization: Optional[str]): if not organization: organization, _ = self.get_organization_project_by_git() - organization_pattern = r"https:\/\/(.*)\.visualstudio\.com" + organization_pattern = r"https:\/\/([^/]+)\.visualstudio\.com" result = re.findall(pattern=organization_pattern, string=organization) if result: organization_name = result[0] else: - organization_pattern = r"https:\/\/dev\.azure\.com\/(.*)" + organization_pattern = r"https:\/\/dev\.azure\.com\/([^/]+)" result = re.findall(pattern=organization_pattern, string=organization) if not result: raise RuntimeError("Cannot find artifact organization.")