Felix Ruess
01/02/2023, 2:40 PMMaarten de Jong
01/02/2023, 4:56 PMclass CustomContainerTask(ContainerTask):
class CustomContainerTask(ContainerTask):
def __init__(
self,
requests: Optional[Resources] = None,
limits: Optional[Resources] = None,
docker_image_tag: str = DOCKER_IMAGE_TAG_LATEST,
**kwargs: Any,
):
super().__init__(
name="custom_container_task",
input_data_dir="/var/inputs",
output_data_dir="/var/outputs",
inputs=kwtypes(file1=FlyteFile, file2=FlyteFile),
image=f"{DOCKER_REGISTRTY_URL}:{docker_image_tag}",
requests=requests,
limits=limits,
metadata_format=ContainerTask.MetadataFormat.JSON,
command=["/bin/sh", "-c"],
arguments=[" ".join(arguments)],
**kwargs,
)
@dynamic
def start_some_custom_task() -> None:
pod_resources = Resources(cpu="3", mem="20Gi", gpu="1")
export_terrain_texture_container_task = ExportTerrainTextureContainerTask(
limits=pod_resources,
requests=pod_resources,
docker_image_tag=docker_image_tag,
)
Felix Ruess
01/02/2023, 5:30 PM{{.input.x}}
e.g. for command is done?Felix Ruess
01/02/2023, 7:14 PMYee
Felix Ruess
01/02/2023, 7:35 PMYee
Felix Ruess
01/02/2023, 7:47 PMYee
Felix Ruess
01/03/2023, 11:48 PMKevin Su
01/12/2023, 10:47 PM