Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add ImageSpec in raw container as example
Browse files Browse the repository at this point in the history
Signed-off-by: machichima <[email protected]>
machichima committed Dec 7, 2024
1 parent 0c8041d commit 7f5da28
Showing 1 changed file with 7 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import logging

from flytekit import ContainerTask, kwtypes, task, workflow
from flytekit import ContainerTask, kwtypes, task, workflow, ImageSpec
from flytekit.core.base_task import TaskMetadata

logger = logging.getLogger(__file__)
@@ -35,7 +35,12 @@
output_data_dir="/var/outputs",
inputs=kwtypes(a=float, b=float),
outputs=kwtypes(area=float, metadata=str),
image="ghcr.io/flyteorg/rawcontainers-python:v2",
image=ImageSpec(
base_image="ghcr.io/flyteorg/rawcontainers-python:v2",
registry="localhost:30000",
builder="default",
copy=["calculate-ellipse-area.py"],
),
command=[
"python",
"calculate-ellipse-area.py",

0 comments on commit 7f5da28

Please sign in to comment.