Skip to content

Commit

Permalink
refactor: Rename image for improved container ux
Browse files Browse the repository at this point in the history
Rename `container_image` to `image` in the example code

Signed-off-by: JiangJiaWei1103 <waynechuang97@gmail.com>
  • Loading branch information
JiangJiaWei1103 committed Feb 4, 2025
1 parent 02e0dc9 commit 39047a7
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from flytekit import task, workflow


@task(container_image="{{.image.mindmeld.fqn}}:{{.image.mindmeld.version}}")
@task(image="{{.image.mindmeld.fqn}}:{{.image.mindmeld.version}}")
def get_data() -> np.ndarray:
# here we're importing scikit learn within the Flyte task
from sklearn import datasets
Expand All @@ -12,7 +12,7 @@ def get_data() -> np.ndarray:
return X


@task(container_image="{{.image.borebuster.fqn}}:{{.image.borebuster.version}}")
@task(image="{{.image.borebuster.fqn}}:{{.image.borebuster.version}}")
def normalize(X: np.ndarray) -> np.ndarray:
return (X - X.mean(axis=0)) / X.std(axis=0)

Expand Down

0 comments on commit 39047a7

Please sign in to comment.