[ci][byod/ml][1] migrate some ml tests to byod

[can-anyscale]

Why are these changes needed?

Release test infra now support running tests using anyscale byod, which is a more reliable way of testing.

• Migrate some ml tests to byod

Checks

• [X] I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
• Testing Strategy
  • [X] Unit tests
  • [X] Release tests - https://buildkite.com/ray-project/release-tests-pr/builds/42221 (failed tests are not byod)