Max results when parent job is specificed (#33830)

Author: nemanjarajic

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

@@ -287,10 +287,11 @@ def list(
 
         schedule_defined = kwargs.pop("schedule_defined", None)
         scheduled_job_name = kwargs.pop("scheduled_job_name", None)
+        max_results = kwargs.pop("max_results", None)
 
         if parent_job_name:
             parent_job = self.get(parent_job_name)
-            return self._runs_operations.get_run_children(parent_job.name)
+            return self._runs_operations.get_run_children(parent_job.name, max_results=max_results)
 
         return cast(
             Iterable[Job],

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

@@ -44,14 +44,15 @@ def get_run_details(self, run_id: str) -> RunDetails:
             run_id,
         )
 
-    def get_run_children(self, run_id: str) -> Iterable[_BaseJob]:
+    def get_run_children(self, run_id: str, **kwargs) -> Iterable[_BaseJob]:
         return cast(
             Iterable[_BaseJob],
             self._operation.get_child(
                 self._subscription_id,
                 self._resource_group_name,
                 self._workspace_name,
                 run_id,
+                top=kwargs.pop("max_results", None),
                 cls=lambda objs: [self._translate_from_rest_object(obj) for obj in objs],
             ),
         )