def _main()

in src/utils/compile_utils.py [0:0]

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

def _main(argv):
    """Compiles the kubeflow pipeline"""
    ip_address, network = get_filestore_info(FLAGS.project_id,
                                              FLAGS.filestore_instance_id,
                                              FLAGS.filestore_instance_location)

    os.environ['ALPHAFOLD_COMPONENTS_IMAGE'] = FLAGS.alphafold_components_image
    os.environ['NFS_SERVER'] = ip_address
    os.environ['NFS_PATH'] = FLAGS.filestore_share
    os.environ['NETWORK'] = network
    os.environ['MODEL_PARAMS_GCS_LOCATION'] = FLAGS.model_params_path
    os.environ['DATA_PIPELINE_MACHINE_TYPE'] = FLAGS.data_pipeline_machine_type

    if FLAGS.predict_gpu == 'nvidia-tesla-a100':
        os.environ['MEMORY_LIMIT'] = '85'
        os.environ['CPU_LIMIT'] = '12'
        os.environ['GPU_LIMIT'] = '1'
        os.environ['GPU_TYPE'] = 'nvidia-tesla-a100'

    if FLAGS.relax_gpu == 'nvidia-tesla-a100':
        os.environ['RELAX_MEMORY_LIMIT'] = '85'
        os.environ['RELAX_CPU_LIMIT'] = '12'
        os.environ['RELAX_GPU_LIMIT'] = '1'
        os.environ['RELAX_GPU_TYPE'] = 'nvidia-tesla-a100'

    pipeline_func, _ = _get_fun_by_name(FLAGS.pipeline_fun)
    compiler.Compiler().compile(
        pipeline_func=pipeline_func,
        package_path=FLAGS.pipeline_template_path
    )