Op in sub-graph getting passed a InputMappingNode object?

[whillas-yabble]

Hi,

I'm trying to break my graph up into two sub graphs with some conditional calling like so:

from dagster import graph, op, Output, Out
from random import randint

@op(
    out={
        "some_data": Out(is_required=False),
        "no_data_found": Out(is_required=False),
    }
)
def get_some_data(data_id: int):
    if randint(0, 1) == 0:
        yield Output(None, "no_data_found")
    else:
        yield Output({"yup": "data"}, "some_data") # etc

@op
def do_something_op1(subject: dict):
    return list(subject.values())

@graph
def child_graph1(subject):
    return do_something_op1(subject)

@graph
def child_graph2(subject) -> dict:
    ...

@graph
def mother():

    data, nope = get_some_data(1) # only one of these is returned
    result = child_graph1(nope)
    result = child_graph2(data)

    return result

So do_something_op1 is pulling the error:

AttributeError: 'InputMappingNode' object has no attribute 'values'

Why is it getting a InputMappingNode instead of a dict? How can i fix this?

[whillas-yabble]

Forgot to put an @op on the function :-?