# PIPELINE DEFINITION
# Name: sample-pipeline
# Inputs:
#    batch_size: int
components:
  comp-train:
    executorLabel: exec-train
    inputDefinitions:
      parameters:
        batch_size:
          parameterType: NUMBER_INTEGER
    outputDefinitions:
      artifacts:
        model:
          artifactType:
            schemaTitle: system.Model
            schemaVersion: 0.0.1
      parameters:
        batch_size:
          parameterType: NUMBER_INTEGER
deploymentSpec:
  executors:
    exec-train:
      container:
        args:
          - echo 'I am a model' > {{$.outputs.artifacts['model'].path}}
        command:
          - sh
          - -c
        image: python:3.10
pipelineInfo:
  name: sample-pipeline
root:
  dag:
    tasks:
      train:
        cachingOptions:
          enableCache: true
        componentRef:
          name: comp-train
        inputs:
          parameters:
            batch_size:
              componentInputParameter: batch_size
        taskInfo:
          name: train
  inputDefinitions:
    parameters:
      batch_size:
        parameterType: NUMBER_INTEGER
schemaVersion: 2.1.0
sdkVersion: kfp-2.4.0