Skip to content

Commit

Permalink
updated readme
Browse files Browse the repository at this point in the history
Updated readme to include section about troubleshooting issues with the Vertex AI Custom Code Service Agent
  • Loading branch information
theo-dt committed Nov 27, 2024
1 parent dff22d4 commit 9be489b
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -164,6 +164,12 @@ make test [ packages=<pipelines components> ]

For details on setting up CI/CD, see [this guide](./docs/Automation.md).

## Issues with Vertex AI Custom Code Service Agent

If you run custom training code to train a custom-trained model, then the [Vertex AI Custom Code Service Agent](https://cloud.google.com/vertex-ai/docs/general/access-control) will be used. In those cases, the agent is created only when you first try to run custom training code which means you can't assign permissions to the agent, like artifact registry reader, from the very beginning. To tackle this, you can use [this guide](https://github.com/teamdatatonic/terraform-google-vertex-cc-service-agent). This repo uses the curl command to create a simple custom training job which triggers the creation of the service agent. You can also edit that code to use ```gcloud ai custom-jobs create``` to create the job if you want.

Alternatively, Google has another method of triggering the creation of service agents that is currently in pre-ga that can be used instead of the above solution. You can read more about it [here](https://cloud.google.com/iam/docs/create-service-agents#create)

## Putting it all together

For a full walkthrough of the journey from changing the ML pipeline code to having it scheduled and running in production, please see the guide [here](./docs/Production.md).
Expand Down

0 comments on commit 9be489b

Please sign in to comment.