chore: Fix unit tests#6737

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

copybara_910589465googleapis/python-aiplatform:copybara_910589465Copy head branch name to clipboard