Function as_apply

hyperopt/pyll/base.py:196–219

Smart way of turning object into an Apply

(obj)

def as_apply(obj):
    """Smart way of turning object into an Apply"""
    if isinstance(obj, Apply):
        rval = obj
    elif isinstance(obj, tuple):
        rval = Apply("pos_args", [as_apply(a) for a in obj], {}, len(obj))
    elif isinstance(obj, list):
        rval = Apply("pos_args", [as_apply(a) for a in obj], {}, None)
    elif isinstance(obj, dict):
        items = list(obj.items())
        # -- should be fine to allow numbers and simple things
        #    but think about if it's ok to allow Apply objects
        #    it messes up sorting at the very least.
        items.sort()
        if all(isinstance(k, str) for k in obj):
            named_args = [(k, as_apply(v)) for (k, v) in items]
            rval = Apply("dict", [], named_args, len(named_args))
        else:
            new_items = [(k, as_apply(v)) for (k, v) in items]
            rval = Apply("dict", [as_apply(new_items)], {}, o_len=None)
    else:
        rval = Literal(obj)
    assert isinstance(rval, Apply)
    return rval


class Apply: