Simple Papermill Notebook

This notebook is used in the previous example as a NotebookTask to demonstrate how to use the papermill plugin.

The cell below has the parameters tag, defining the inputs to the notebook.

v = 3.14

Then, we do some computation:

square = v*v
print(square)

Finally, we use the flytekitplugins.papermill package to record the outputs so that flyte understands which state to serialize and pass into a downstream task.

from flytekitplugins.papermill  import record_outputs

record_outputs(square=square)