{
   "components": {
     "comp-notebooks-executor": {
       "executorLabel": "exec-notebooks-executor",
       "inputDefinitions": {
         "parameters": {
           "accelerator_core_count": {
             "defaultValue": "0",
             "isOptional": true,
             "parameterType": "STRING"
           },
           "accelerator_type": {
             "isOptional": true,
             "parameterType": "STRING"
           },
           "block_pipeline": {
             "defaultValue": true,
             "isOptional": true,
             "parameterType": "BOOLEAN"
           },
           "container_image_uri": {
             "parameterType": "STRING"
           },
           "execution_id": {
             "parameterType": "STRING"
           },
           "fail_pipeline": {
             "defaultValue": true,
             "isOptional": true,
             "parameterType": "BOOLEAN"
           },
           "input_notebook_file": {
             "parameterType": "STRING"
           },
           "job_type": {
             "defaultValue": "VERTEX_AI",
             "isOptional": true,
             "parameterType": "STRING"
           },
           "kernel_spec": {
             "defaultValue": "python3",
             "isOptional": true,
             "parameterType": "STRING"
           },
           "labels": {
             "defaultValue": "src=notebooks_executor_api",
             "isOptional": true,
             "parameterType": "STRING"
           },
           "location": {
             "parameterType": "STRING"
           },
           "master_type": {
             "parameterType": "STRING"
           },
           "output_notebook_folder": {
             "parameterType": "STRING"
           },
           "parameters": {
             "isOptional": true,
             "parameterType": "STRING"
           },
           "params_yaml_file": {
             "isOptional": true,
             "parameterType": "STRING"
           },
           "project": {
             "parameterType": "STRING"
           },
           "service_account": {
             "isOptional": true,
             "parameterType": "STRING"
           }
         }
       },
       "outputDefinitions": {
         "parameters": {
           "error": {
             "parameterType": "STRING"
           },
           "gcp_resources": {
             "parameterType": "STRING"
           },
           "output_notebook_file": {
             "parameterType": "STRING"
           },
           "state": {
             "parameterType": "STRING"
           }
         }
       }
     }
   },
   "deploymentSpec": {
     "executors": {
       "exec-notebooks-executor": {
         "container": {
           "args": [
             "--project",
             "{{$.inputs.parameters['project']}}",
             "--input_notebook_file",
             "{{$.inputs.parameters['input_notebook_file']}}",
             "--output_notebook_folder",
             "{{$.inputs.parameters['output_notebook_folder']}}",
             "--execution_id",
             "{{$.inputs.parameters['execution_id']}}",
             "--location",
             "{{$.inputs.parameters['location']}}",
             "--master_type",
             "{{$.inputs.parameters['master_type']}}",
             "--container_image_uri",
             "{{$.inputs.parameters['container_image_uri']}}",
             "{\"IfPresent\": {\"InputName\": \"accelerator_type\", \"Then\": [\"--accelerator_type\", \"{{$.inputs.parameters['accelerator_type']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"accelerator_core_count\", \"Then\": [\"--accelerator_core_count\", \"{{$.inputs.parameters['accelerator_core_count']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"labels\", \"Then\": [\"--labels\", \"{{$.inputs.parameters['labels']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"params_yaml_file\", \"Then\": [\"--params_yaml_file\", \"{{$.inputs.parameters['params_yaml_file']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"parameters\", \"Then\": [\"--parameters\", \"{{$.inputs.parameters['parameters']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"service_account\", \"Then\": [\"--service_account\", \"{{$.inputs.parameters['service_account']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"job_type\", \"Then\": [\"--job_type\", \"{{$.inputs.parameters['job_type']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"kernel_spec\", \"Then\": [\"--kernel_spec\", \"{{$.inputs.parameters['kernel_spec']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"block_pipeline\", \"Then\": [\"--block_pipeline\", \"{{$.inputs.parameters['block_pipeline']}}\"]}}",
             "{\"IfPresent\": {\"InputName\": \"fail_pipeline\", \"Then\": [\"--fail_pipeline\", \"{{$.inputs.parameters['fail_pipeline']}}\"]}}",
             "----output-paths",
             "{{$.outputs.parameters['state'].output_file}}",
             "{{$.outputs.parameters['output_notebook_file'].output_file}}",
             "{{$.outputs.parameters['gcp_resources'].output_file}}",
             "{{$.outputs.parameters['error'].output_file}}"
           ],
           "command": [
             "python3",
             "-m",
             "google_cloud_pipeline_components.container.experimental.notebooks.executor"
           ],
           "image": "gcr.io/ml-pipeline/google-cloud-pipeline-components:2.0.0b1"
         }
       }
     }
   },
   "pipelineInfo": {
     "name": "notebooks-executor-test"
   },
   "root": {
     "dag": {
       "tasks": {
         "notebooks-executor": {
           "cachingOptions": {
             "enableCache": true
           },
           "componentRef": {
             "name": "comp-notebooks-executor"
           },
           "inputs": {
             "parameters": {
               "container_image_uri": {
                 "runtimeValue": {
                   "constant": "gcr.io/deeplearning-platform-release/base-cpu"
                 }
               },
               "execution_id": {
                 "runtimeValue": {
                   "constant": "notebook_executor_test"
                 }
               },
               "input_notebook_file": {
                 "runtimeValue": {
                   "constant": "test_notebook_file"
                 }
               },
               "location": {
                 "runtimeValue": {
                   "constant": "us-central1"
                 }
               },
               "master_type": {
                 "runtimeValue": {
                   "constant": "n1-standard-4"
                 }
               },
               "output_notebook_folder": {
                 "runtimeValue": {
                   "constant": "test_output_folder"
                 }
               },
               "parameters": {
                 "runtimeValue": {
                   "constant": "PROJECT_ID=test_project,EXECUTION_ID=notebook_executor_test"
                 }
               },
               "project": {
                 "runtimeValue": {
                   "constant": "test_project"
                 }
               }
             }
           },
           "taskInfo": {
             "name": "notebooks-executor"
           }
         }
       }
     }
   }
 }