BUG: Dictionaries with output references as keys do not resolve

[Andrew-S-Rosen]

In the following code, we expect the output references to be resolved, but they are not. An OutputReference appears in the response. Everything works as expected when the value has an OutputReference but not when the key does.

import jobflow as jf


@jf.job
def test(d: dict):
    d["test"] = "text"
    return d

job1 = test({"a": "b"})
job2 = test({job1.output["test"]: "c"})
responses = jf.run_locally(jf.Flow([job1, job2]), ensure_success=True)
print(responses[job2.uuid][1].output)

{OutputReference(938990aa-3c5e-4d7a-97ae-430dfac74b4d, ['test']): 'c', 'test': 'text'}

[Andrew-S-Rosen]

It looks like there are no tests for this in

jobflow/tests/core/test_reference.py

Line 364 in f24cfca

def test_find_and_resolve_references(memory_jobstore):

. It's only testing against values, not keys.

[utf]

Thanks for noticing this @Andrew-S-Rosen. I remember running into this issue when I was initially writing jobflow and deciding not to implement it. I can't remember the reason for that now. I agree it would be nice if this was supported.

[Andrew-S-Rosen]

I think this one proved more difficult than originally anticipated, so @vineetbansal and I decided not to pursue it in the short term.