chore: fix tensorboard cleanup in system tests

PiperOrigin-RevId: 510180069

Author: sararob

tests/system/aiplatform/e2e_base.py

@@ -179,6 +179,7 @@ def tear_down_resources(self, shared_state: Dict[str, Any]):
             key=lambda r: 1
             if isinstance(r, aiplatform.Endpoint)
             or isinstance(r, aiplatform.MatchingEngineIndexEndpoint)
+            or isinstance(r, aiplatform.Experiment)
             else 2
         )
 
@@ -201,12 +202,3 @@ def tear_down_resources(self, shared_state: Dict[str, Any]):
                     resource.delete()
             except exceptions.GoogleAPIError as e:
                 logging.error(f"Could not delete resource: {resource} due to: {e}")
-
-        # When an Experiment has a backing_tensorboard, the Experiment needs to be deleted first
-        # This is used by the autologging tests
-        if "tensorboard" in shared_state:
-            for resource in shared_state["tensorboard"]:
-                try:
-                    resource.delete()
-                except exceptions.GoogleAPIError as e:
-                    logging.error(f"Could not delete resource: {resource} due to: {e}")

tests/system/aiplatform/test_autologging.py

@@ -152,11 +152,11 @@ def test_autologging_with_autorun_creation(self, shared_state):
             experiment_tensorboard=self._backing_tensorboard,
         )
 
-        shared_state["tensorboard"] = [self._backing_tensorboard]
+        shared_state["resources"] = [self._backing_tensorboard]
 
-        shared_state["resources"] = [
+        shared_state["resources"].append(
             aiplatform.metadata.metadata._experiment_tracker.experiment
-        ]
+        )
 
         aiplatform.autolog()