thankful-minister-83577
faint-rocket-14513
03/15/2024, 12:39 AMfrom flytekit import task, Resources # Define resource requirements for each domain RES = { "development": { "requests": Resources(cpu="1", mem="100Mi"), "limits": Resources(cpu="2", mem="200Mi"), }, "production": { "requests": Resources(cpu="2", mem="500Mi"), "limits": Resources(cpu="4", mem="1Gi"), }, } # Use the domain-specific resource requirements in your task @task(requests=RES[settings.domain]["requests"], limits=RES[settings.domain]["limits"]) def my_task(): ...
In this example, settings.domain
would be a variable that specifies the current domain (e.g., "development" or "production"). This allows the my_task
function to use different resource specifications depending on the domain it's being registered or executed in. This method provides flexibility in managing resource requirements across different environments, ensuring that tasks have the appropriate resources allocated based on the domain's needs.