AzureML: Fix ARM ID for registry environments (Azure#27305)

* Fix ARM id retrieval for registry environments

* Update changelog

* Update test_operation_orchestrator.py

Author: Man-MSFT

sdk/ml/azure-ai-ml/CHANGELOG.md

@@ -17,6 +17,7 @@
 - Adjust registry experimental tags and imports to avoid warning printouts for unrelated operations.
 - Make registry delete operation return an LROPoller, and change name to begin_delete.
 - Prevent registering an already existing environment that references conda file.
+- Fix ARM id logic for registry environments (ex: Creating a registry component that references a registry environment).
 
 ### Other Changes
 

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

@@ -40,6 +40,7 @@
     AzureMLResourceType,
     LABELLED_RESOURCE_ID_FORMAT,
     DEFAULT_LABEL_NAME,
+    REGISTRY_VERSION_PATTERN
 )
 from azure.ai.ml.entities import Component
 from azure.ai.ml.entities._assets import Code, Data, Environment, Model
@@ -141,7 +142,7 @@ def get_asset_arm_id(
                 # CLI strips off azureml: in the schema, appending it back as required by backend
                 if (
                     azureml_type == "environments"
-                    and asset.startswith(CURATED_ENV_PREFIX)
+                    and (asset.startswith(CURATED_ENV_PREFIX) or re.match(REGISTRY_VERSION_PATTERN, f"azureml:{asset}"))
                 ):
                     return f"azureml:{asset}"
 
@@ -181,7 +182,7 @@ def get_asset_arm_id(
                             "Use fully qualified name to reference custom environments "
                             "when creating assets in registry. "
                             "The syntax for fully qualified names is "
-                            "azureml:registries/azureml/environments/{{env-name}}/versions/{{version}}"
+                            "azureml://registries/azureml/environments/{{env-name}}/versions/{{version}}"
                         )
                         raise ValidationException(
                             message=msg.format(asset, azureml_type),

sdk/ml/azure-ai-ml/tests/internal_utils/unittests/test_operation_orchestrator.py

@@ -186,6 +186,12 @@ def create_yaml_inline_model(tmp_path: Path, resource_group_name: str) -> Path:
 @pytest.mark.unittest
 @pytest.mark.core_sdk_test
 class TestOperationOrchestration:
+    def test_registry_environment(self, operation_orchestrator: OperationOrchestrator) -> None:
+        test_input = '//registries/mafong-registry/environments/conda_name_version_e2e/versions/1.0.2'
+        expected = 'azureml://registries/mafong-registry/environments/conda_name_version_e2e/versions/1.0.2'
+        actual = operation_orchestrator.get_asset_arm_id(test_input, azureml_type=AzureMLResourceType.ENVIRONMENT)
+        assert actual == expected
+
     def test_code_arm_id(self, operation_orchestrator: OperationOrchestrator) -> None:
         code = VERSIONED_RESOURCE_ID_FORMAT.format(
             Test_Subscription,