Support capacity provider strategy in task run request

[kaaloo]

This PR allows setting a capacityProviderStrategy in the task_run_request part of an ECS Task template. It works by checking for the capacityProviderStrategy key when preparing the run request and dropping the launchType key since these are mutually exclusive. See:

https://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_RunTask.html#ECS-RunTask-request-capacityProviderStrategy

Closes PrefectHQ/prefect#13030

Example

The following Gist illustrates an AWS Push Work Pool template that has been modified to include a new capacityProvider variable, which is then used in the capacityProviderStrategy section of the task_run_template.

Screenshots

Checklist

• References any related issue by including "Closes #" or "Closes ".
  • If no issue exists and your change is not a small fix, please create an issue first.
• Includes tests or only affects documentation.
• Passes pre-commit checks.
  • Run pre-commit install && pre-commit run --all locally for formatting and linting.
• Includes screenshots of documentation updates.
  • Run mkdocs serve view documentation locally.
• Summarizes PR's changes in CHANGELOG.md

[jakekaplan]

Apologies for the delay on this, I have not had time to test it. Will try to do so ASAP

[jggatter]

Looking forward to this, thank you @kaaloo!

[jakekaplan]

Thanks for the contribution! Apologies for the delay in the review. Approved barring any necessary changes to make to pass CI.

• It seems you might have a different black config:
  https://github.com/PrefectHQ/prefect-aws/actions/runs/6173273421/job/17051020965?pr=312
• Additionally it seems like the test is failing:
  https://github.com/PrefectHQ/prefect-aws/actions/runs/6173273411/job/17051021768?pr=312

[kaaloo]

Hello @jakekaplan , I updated my PR. Could you please approve running the workflows?

[jakekaplan]

Hey it looks like the tests are still failing unfortunately. If you're able to fix that would be great! I can try and take a look myself if I have sometime.

[EmilRex]

@kaaloo Thank you again for putting in this PR! Just suggested a couple of changes that should fix the linting issues and a question about the test you've implemented.

[EmilRex]

This seems like the right approach but the tests are failing saying that capacityProviderName is None. Maybe it's not a value that's mocked correctly? The value is supposed to be there according to the spec. Perhaps it's not coming through because it's not one of the accepted values? @kaaloo was the rest passing for you locally?

[anhpham1509]

I'd like to chime in here and see that mock ecs cluster is defined here without capacity provider created. That might be the issue that it's None here in this test case.

[desertaxle]

Thank you for your willingness to contribute to prefect-aws! The code for this library has been moved to https://github.com/PrefectHQ/prefect/tree/main/src/integrations/prefect-aws. Please reopen this PR there if this functionality is still needed!