def _maybe_merge_inputs(inputs): if isinstance(inputs, list) and len(inputs) > 1: return Merge(mode='concat')(inputs) elif isinstance(inputs, list) and len(inputs) == 1: return inputs[0] else: return inputs