def ensure_tf_input()

in maddpg/common/tf_util.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def ensure_tf_input(thing):
    """Takes either tf.placeholder of TfInput and outputs equivalent TfInput"""
    if isinstance(thing, TfInput):
        return thing
    elif is_placeholder(thing):
        return PlacholderTfInput(thing)
    else:
        raise ValueError("Must be a placeholder or TfInput")