Does anybody else run into errors when trying to use map task workflows with Papermill/Jupyter NotebookTasks? Running the example from the docs (https://docs.flyte.org/projects/cookbook/en/latest/auto/core/control_flow/map_task.html#map-a-task-with-multiple-inputs), and a plain @task decorated function works fine - but the moment I attempt to run a map task with NotebookTasks, I get this error:

ValueError: Error encountered while executing 'run_map_task_workflow':
  Map tasks can only compose of Python Functon Tasks currently

Is

flytekitplugins.papermill.NotebookTask

currently incompatible with map_tasks?

Great point. this is not supported, but actually is very easy to support. can you share how you are wanting to run the notebook task, are you using

functools.partial

looking

this is the problem - @glamorous-carpet-83516 - https://github.com/flyteorg/flytekit/blob/ba70f4685e5ab4e00f23c5adedb281110b20f96e/flytekit/core/map_task.py#L57

@freezing-airport-6809 I'm not too sure about functools.partial, but we are using the flytekit plugin for Jupyter.

*from* flytekitplugins.papermill *import* NotebookTask

And then we are passing in this NotebookTask into something like this:

# Example

@workflow
def map_workflow():
   map_output = map_task(notebook_task_here)(inputs=input_list)
   return map_output

Cool, ya this is a miss on my end when I wrote map task

Cool, thank you for the quick response!

Fix it. https://github.com/flyteorg/flytekit/pull/1650