Update vertex.py with scheduling parameters

[jbrache]

Overview of changes

Expose the ability to specify the scheduling parameter: #15417

Checklist

• [ X ] This pull request references any related issue by including "closes <link to issue>"
  • Closes Expose Vertex Dynamic Workload Scheduler on Vertex Run #15417
  • Closes VertexAI workpools Scheduling args #17277
• [ X ] If this pull request adds new functionality, it includes unit tests that cover the changes

[desertaxle]

Thanks for the PR @jbrache! Could you please write some tests to cover this new functionality? Let me know if you need any help with that and I can provide some guidance.

[jbrache]

Thanks for the PR @jbrache! Could you please write some tests to cover this new functionality? Let me know if you need any help with that and I can provide some guidance.

Hi @desertaxle yea any guidance in providing test coverage, happy to take a look at what you're expecting to see.

[desertaxle]

@jbrache a test like this one would work well. You would create a new fixture containing a job configuration that uses the new scheduling parameter and assert that create_custom_job is called with the expected arguments. The test I linked is a good example of how to perform such assertions on the mocked client.

[jbrache]

Added tests in test_vertex_worker.py including a new fixture containing a job configuration that uses the new scheduling parameter. I could have just re-used the existing job_config() test fixture and if you would like I can just re-use it to clean it up a bit.

[zzstoatzz]

thanks for the PR @jbrache ! a nit and a question so far from me

[zzstoatzz]

2 more small nits but otherwise this is looking good to me!

[zzstoatzz]

thank you for the PR @jbrache !

[desertaxle]

LGTM! Thank you!