The main challenge is the task

compute_cop

which has two dependencies, and when I try to replicate that with

Flow.add_edge()

I can only specify a single

key

to pass the data between tasks. Does anyone know how to get solve or get around this?

I believe you can do

cop.set_upstream([task1, task2, ...])

to set dependencies. But if

result

and

power

are the dependencies you're referring to, prefect should automatically be able to identify them as dependencies since they are inputs to the

compute_cop

task. You can use

flow.visualize()

to see the dag and verify the dependencies.

set_upstream

seems to be a good suggestion since it's a

Task

method but it also takes an optional

flow

parameter. I'll give it go, thanks!

The flow parameter is detected from the context if it’s not passed — so as long as you’re in the

with Flow ….

you don’t need to provide it.

Sure, that works, but in this case I'm trying to specify the flow using the imperative API:

flow = Flow(myflow)
flow.add_task(fetch_dataframe)
...