Thanks for the question! Clarification -- by "yield" do you mean "return a single function's result" or "use a function that's a generator"?

In the former case, yes, it's pretty easy:

  driver.execute(final_vars=["my_func_1", "my_func_2"]) 

will just give out the specific ones you want. In the latter case, it's a little trickier but doable -- we were just going over this with a user recently actually! https://github.com/DAGWorks-Inc/hamilton/issues/90