diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_model_operations.py b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_model_operations.py index 625aa837fcda..d589b945f3c3 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_model_operations.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/operations/_model_operations.py @@ -629,6 +629,7 @@ def list( self._registry_name, ArmModelVersion, Model._from_rest_object, + list_view_type=list_view_type, ) if self._registry_name else self._model_versions_operation.list( diff --git a/sdk/ml/azure-ai-ml/tests/model/unittests/test_model_operations.py b/sdk/ml/azure-ai-ml/tests/model/unittests/test_model_operations.py index db6657fc558b..6dee26850b6b 100644 --- a/sdk/ml/azure-ai-ml/tests/model/unittests/test_model_operations.py +++ b/sdk/ml/azure-ai-ml/tests/model/unittests/test_model_operations.py @@ -6,6 +6,7 @@ from azure.ai.ml import load_model from azure.ai.ml._restclient.arm_ml_service.models import ( + ListViewType, ModelContainer as ModelContainerData, ModelContainerProperties as ModelContainerDetails, ModelVersion as ModelVersionData, @@ -268,6 +269,14 @@ def test_list(self, mock_model_operation: ModelOperations) -> None: mock_model_operation.list(name="random_string") mock_model_operation._model_versions_operation.list.assert_called_once() + def test_list_registry_versions_forwards_list_view_type( + self, mock_model_operation_reg: ModelOperations + ) -> None: + with patch("azure.ai.ml.operations._model_operations.list_registry_assets") as mock_list: + mock_model_operation_reg.list(name="random_string", list_view_type=ListViewType.ARCHIVED_ONLY) + + assert mock_list.call_args.kwargs["list_view_type"] == ListViewType.ARCHIVED_ONLY + def test_archive_version(self, mock_model_operation: ModelOperations) -> None: name = "random_string" model_version = Mock(ModelVersionData(properties=Mock(ModelVersionDetails())))