chore: Fix flaky unit tests#6740

copybara-service[bot] wants to merge 1 commit intomaingoogleapis/python-aiplatform:mainfrom

copybara_910960736googleapis/python-aiplatform:copybara_910960736Copy head branch name to clipboard