"""
pip install git+<https://github.com/PrefectHQ/prefect-aws.git@main>
prefect block register -m prefect_aws.ecs
"""
from prefect_aws.ecs import ECSTask, AwsCredentials

aws_credentials_block = AwsCredentials.load("prod")

ecs = ECSTask(
    command=["echo", "hello ECS"],
    aws_credentials=aws_credentials_block,
    image="prefecthq/prefect:2-python3.9",
    cpu="256",
    memory="512",
    stream_output=True,
    configure_cloudwatch_logs=True,
    cluster="prefect",
    vpc_id="your_private_vpc",
    execution_role_arn="arn:aws:iam::1111111111:role/dataflowops_ecs_execution_role",
    task_role_arn="arn:aws:iam::11111111111:role/dataflowops_ecs_execution_role",
    task_start_timeout_seconds=90
)
await ecs.run()