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).

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

)

have you installed the plugin in the backend? https://docs.flyte.org/en/latest/deployment/plugins/k8s/index.html