def clean_up_pipeline_args()

in spotify_tensorflow/tfx/utils.py [0:0]


def clean_up_pipeline_args(pipeline_args):
    output_args = list()
    for arg in pipeline_args:
        if arg.startswith("--"):
            if "=" in arg:
                k, v = arg.split("=")
                output_args.extend([to_snake_case(k), v])
            else:
                output_args.append(to_snake_case(arg))
        else:
            output_args.append(arg)

    keys = output_args[0::2]
    vals = output_args[1::2]
    return ["%s=%s" % (key, val) for (key, val) in zip(keys, vals)
            if key in SUPPORTED_DATAFLOW_PIPELINE_ARGS]