def main()

in basic_pitch/data/datasets/slakh.py [0:0]

• 26 lines of code
• 1 McCabe index (conditional complexity)

def main(known_args: argparse.Namespace, pipeline_args: List[str]) -> None:
    time_created = int(time.time())
    destination = commandline.resolve_destination(known_args, time_created)
    input_data = create_input_data()

    pipeline_options = {
        "runner": known_args.runner,
        "job_name": f"slakh-tfrecords-{time_created}",
        "machine_type": "e2-standard-4",
        "num_workers": 25,
        "disk_size_gb": 128,
        "experiments": ["use_runner_v2"],
        "save_main_session": True,
        "sdk_container_image": known_args.sdk_container_image,
        "job_endpoint": known_args.job_endpoint,
        "environment_type": "DOCKER",
        "environment_config": known_args.sdk_container_image,
    }
    pipeline.run(
        pipeline_options,
        pipeline_args,
        input_data,
        SlakhToTfExample(known_args.source, download=True),
        SlakhFilterInvalidTracks(known_args.source),
        destination,
        known_args.batch_size,
    )