https://flyte.org logo
#ask-the-community
Title
# ask-the-community
a

Alykhan Tejani

02/05/2024, 4:47 PM
Hi all, Im trying to set up a dev environment to make some changes to plugin
flyteplugins-ktensorflow
I am launching the local flyte demo sandbox with local k8s etc. I have the following worflow, which launches remotely and succeeds but doesn't actually launch a tfjob (just a single pod that runs the task).
Copy code
import os

from flytekit import task, workflow
from flytekitplugins.kftensorflow import TfJob, Chief, Worker, PS


@task(
    task_config=TfJob(
        chief=Chief(replicas=1),
        ps=PS(replicas=1),
        worker=Worker(replicas=1),
    ),
)
def launch(name: str) -> None:
    print(os.environ)


@workflow
def tf_job_test() -> None:
    launch(name="local")
am I doing something wrong. Command to run is
pyflyte run --remote --image image.yaml ./tfjob_test.py tf_job_test
where
image.yaml
just contains the deps (
flyteplugins.ktensorflow
)
s

Samhita Alla

02/07/2024, 4:56 PM
have you installed the plugin in the backend? https://docs.flyte.org/en/latest/deployment/plugins/k8s/index.html