chore: Add filtering for experiment.list method

PiperOrigin-RevId: 782967412

Author: ryanjung-jm

google/cloud/aiplatform/metadata/experiment_resources.py

@@ -316,6 +316,7 @@ def get_or_create(
     def list(
         cls,
         *,
+        filter: Optional[str] = None,
         project: Optional[str] = None,
         location: Optional[str] = None,
         credentials: Optional[auth_credentials.Credentials] = None,
@@ -327,6 +328,8 @@ def list(
         ```
 
         Args:
+            filter (str):
+                Optional. A query to filter available resources for matching results.
             project (str):
                 Optional. Project to list these experiments from. Overrides project set in
                 aiplatform.init.
@@ -343,6 +346,8 @@ def list(
         filter_str = metadata_utils._make_filter_string(
             schema_title=constants.SYSTEM_EXPERIMENT
         )
+        if filter:
+            filter_str = f"{filter_str} AND ({filter})"
 
         with _SetLoggerLevel(resource):
             experiment_contexts = context.Context.list(

tests/system/aiplatform/test_experiments.py

@@ -121,6 +121,35 @@ def test_get_run(self):
         assert run.name == _RUN
         assert run.state == aiplatform.gapic.Execution.State.RUNNING
 
+    def test_list_experiment(self):
+        experiments = aiplatform.Experiment.list(
+            project=e2e_base._PROJECT,
+            location=e2e_base._LOCATION,
+        )
+        assert isinstance(experiments, list)
+        assert any(
+            experiment.name == self._experiment_name for experiment in experiments
+        )
+
+    def test_list_experiment_filter(self):
+        experiments = aiplatform.Experiment.list(
+            filter=f"display_name = {self._experiment_name}",
+            project=e2e_base._PROJECT,
+            location=e2e_base._LOCATION,
+        )
+        assert len(experiments) == 1
+        assert any(
+            experiment.name == self._experiment_name for experiment in experiments
+        )
+
+    def test_list_experiment_filter_no_results(self):
+        experiments = aiplatform.Experiment.list(
+            filter="display_name = not_mathcing_filter_name",
+            project=e2e_base._PROJECT,
+            location=e2e_base._LOCATION,
+        )
+        assert len(experiments) == 0
+
     def test_log_params(self):
         aiplatform.init(
             project=e2e_base._PROJECT,