chore: pin google-cloud-aiplatform>=1.61 and fix tests
#454
| Job | Run time |
|---|---|
| 3m 52s | |
| 3m 44s | |
| 3m 11s | |
| 3m 55s | |
| 5m 47s | |
| 5m 40s | |
| 5m 44s | |
| 5m 8s | |
| 2m 25s | |
| 2m 0s | |
| 2m 8s | |
| 2m 51s | |
| 46m 25s |