allow subscription-only requests for registry list operation

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

@@ -15,6 +15,7 @@
 - Dropped support for Python 3.6. The Python versions supported for this release are 3.7-3.10.
 
 ### Features Added
+- Registry list operation now accepts scope value to allow subscription-only based requests.
 
 ### Breaking Changes
 - OnlineDeploymentOperations.delete has been renamed to begin_delete.
@@ -37,6 +38,7 @@
  - Added a `show_progress` parameter to MLClient for enable/disable progress bars of long running operations.
  - Support `month_days` in `RecurrencePattern` when using `RecurrenceSchedule`.
  - Support `ml_client.schedules.list` with `list_view_type`, default to `ENABLED_ONLY`.
+ - Registry list operation now accepts scope value to allow subscription-only based requests.
  - Add support for model sweeping and hyperparameter tuning in AutoML NLP jobs.
  - Added `ml_client.jobs.show_services()` operation.
 

sdk/ml/azure-ai-ml/azure/ai/ml/constants/__init__.py

@@ -4,7 +4,7 @@
 
 __path__ = __import__("pkgutil").extend_path(__path__, __name__)
 
-from ._common import AssetTypes, InputOutputModes, ModelType, TimeZone
+from ._common import AssetTypes, InputOutputModes, ModelType, TimeZone, Scope
 from ._component import ParallelTaskType
 from ._deployment import BatchDeploymentOutputAction
 from ._job import (
@@ -38,4 +38,5 @@
     "AcrAccountSku",
     "NlpModels",
     "NlpLearningRateScheduler",
+    "Scope",
 ]

sdk/ml/azure-ai-ml/azure/ai/ml/constants/_common.py

@@ -562,3 +562,8 @@ class RollingRate:
     DAY = "day"
     HOUR = "hour"
     MINUTE = "minute"
+
+
+class Scope:
+    SUBSCRIPTION="subscription"
+    RESOURCE_GROUP="resource_group"

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

@@ -19,7 +19,7 @@
 
 from azure.ai.ml._utils._experimental import experimental
 from .._utils._azureml_polling import AzureMLPolling
-from ..constants._common import LROConfigurations
+from ..constants._common import LROConfigurations, Scope
 
 ops_logger = OpsLogger(__name__)
 module_logger = ops_logger.module_logger
@@ -52,14 +52,19 @@ def __init__(
         self._init_kwargs = kwargs
 
     #@ monitor_with_activity(logger, "Registry.List", ActivityType.PUBLICAPI)
-    def list(self) -> Iterable[Registry]:
+    def list(self, *, scope: str = Scope.RESOURCE_GROUP) -> Iterable[Registry]:
         """List all registries that the user has access to in the current
-        resource group.
+        resource group or subscription.
 
+        :param scope: scope of the listing, "resource_group" or "subscription", defaults to "resource_group"
+        :type scope: str, optional
         :return: An iterator like instance of Registry objects
         :rtype: ~azure.core.paging.ItemPaged[Registry]
         """
-
+        if scope.lower() == Scope.SUBSCRIPTION:
+            return self._operation.list_by_subscription(
+                cls=lambda objs: [Registry._from_rest_object(obj) for obj in objs]
+            )
         return self._operation.list(cls=lambda objs: [Registry._from_rest_object(obj) for obj in objs], \
             resource_group_name=self._resource_group_name)
 

sdk/ml/azure-ai-ml/tests/registry/unittests/test_registry_operations.py

@@ -29,9 +29,18 @@ def mock_registry_operation(
 @pytest.mark.unittest
 class TestRegistryOperations:
     def test_list(self, mock_registry_operation: RegistryOperations) -> None:
+        # Test different input options for the scope value
         mock_registry_operation.list()
         mock_registry_operation._operation.list.assert_called_once()
 
+        mock_registry_operation.list(scope="invalid")
+        assert mock_registry_operation._operation.list.call_count == 2
+        mock_registry_operation._operation.list_by_subscription.assert_not_called()
+
+        mock_registry_operation.list(scope="subscription")
+        assert mock_registry_operation._operation.list.call_count == 2
+        mock_registry_operation._operation.list_by_subscription.assert_called_once()
+
     def test_get(self, mock_registry_operation: RegistryOperations, randstr: Callable[[], str]) -> None:
         mock_registry_operation.get(f"unittest_{randstr('reg_name')}")
         mock_registry_operation._operation.get.assert_called_once()