fix environment archive and restore (Azure#39818)

* neglect environment.properties.stage from archive

* enable environment archive e2e test

* update model archive recording

Author: pdhotems

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

@@ -4,6 +4,7 @@
 ### Bugs Fixed
   - #37464 - Allowing to update a component in register with anonymousEnvironment environment.
   - #39744 - Fixing a serialization issue in RecurrencePattern schedule.
+  - Fix environment archiving and restore
 
 ## 1.25.0 (2025-02-11)
 

sdk/ml/azure-ai-ml/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ml/azure-ai-ml",
-  "Tag": "python/ml/azure-ai-ml_71dd8cba32"
+  "Tag": "python/ml/azure-ai-ml_8d69a348d2"
 }

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

@@ -1017,6 +1017,7 @@ def _archive_or_restore(
             )
         )
         version_resource.properties.is_archived = is_archived
+        version_resource.properties.stage = None
         (  # pylint: disable=expression-not-assigned
             version_operation.begin_create_or_update(
                 name=name,

sdk/ml/azure-ai-ml/tests/environment/e2etests/test_environment.py

@@ -182,7 +182,6 @@ def get_environment_list():
         client.environments.restore(name=name, version=version_archived)
         assert version_archived in get_environment_list()
 
-    @pytest.mark.skip(reason="Task 1791832: Inefficient, possibly causing testing pipeline to time out.")
     def test_environment_archive_restore_container(self, client: MLClient, env_name: Callable[[str], str]) -> None:
         name = env_name("name")
         params_override = [{"name": name}]