FlyteInteractive Jupyter Decorator#

The @jupyter task decorator launches and monitors a Jupyter notebook server.

Usage#

1. Add the @jupyter decorator to a task function definition#

The @jupyter decorator takes the following optional parameters:

  • max_idle_seconds: Optional[int] (default 36000)

  • port: Optional[int] (default 8888)

  • enable: Optional[bool] (default True)

  • notebook_dir: Optional[str] (default “/root”)

  • pre_execute: Optional[Callable] (default None)

  • post_execute: Optional[Callable], (default None)

from flytekit import task, workflow
from flytekitplugins.flyteinteractive import jupyter


@task
@jupyter
def jupyter_task():
    print("opened notebook")


@workflow
def wf():
    jupyter_task()


if __name__ == "__main__":
    print(wf())

2. Connect to the Jupyter notebook server#

You can connect in two ways:

  • (Recommended) Expose a URL on the Flyte console. Set up ingress on the Flyte backend to expose a URL on the Flyte console. Details are to be determined (TBD).

  • Use port forwarding. To use port forwarding, execute the following command:

    $ kubectl port-forward <pod name> <port>

    Then, open a browser and navigate to localhost:<port>, replacing <port> with the port number configured above. You should be presented with the Jupyter notebook interface.