Correct API call for model archive in registry (#41502)

* fixing API call for model container archive for registry Model

* fix broken test cases

---------

Co-authored-by: Copilot <[email protected]>

Author: achauhan-scc

sdk/ml/azure-ai-ml/azure/ai/ml/_utils/_asset_utils.py

@@ -1037,7 +1037,7 @@ def _archive_or_restore(
         )
         container_resource.properties.is_archived = is_archived
         (  # pylint: disable=expression-not-assigned
-            container_operation.create_or_update(
+            container_operation.begin_create_or_update(
                 name=name,
                 resource_group_name=resource_group_name,
                 registry_name=registry_name,

sdk/ml/azure-ai-ml/tests/component/unittests/test_component_operations_registry.py

@@ -10,8 +10,8 @@
     ComponentVersionDetails,
 )
 from azure.ai.ml._scope_dependent_operations import OperationConfig, OperationScope
-from azure.ai.ml.entities._component.command_component import CommandComponent
 from azure.ai.ml.entities._assets.intellectual_property import IntellectualProperty
+from azure.ai.ml.entities._component.command_component import CommandComponent
 from azure.ai.ml.operations import ComponentOperations
 
 from .._util import _COMPONENT_TIMEOUT_SECOND
@@ -120,7 +120,7 @@ def test_archive_container(self, mock_component_operation: ComponentOperations):
         mock_component_operation._container_operation.get.return_value = component
         mock_component_operation.archive(name=name)
 
-        mock_component_operation._container_operation.create_or_update.assert_called_with(
+        mock_component_operation._container_operation.begin_create_or_update.assert_called_with(
             name=name,
             registry_name=mock_component_operation._registry_name,
             body=component,
@@ -133,7 +133,7 @@ def test_restore_container(self, mock_component_operation: ComponentOperations):
         mock_component_operation._container_operation.get.return_value = component
         mock_component_operation.restore(name=name)
 
-        mock_component_operation._container_operation.create_or_update.assert_called_with(
+        mock_component_operation._container_operation.begin_create_or_update.assert_called_with(
             name=name,
             registry_name=mock_component_operation._registry_name,
             body=component,