diff --git a/sample.yaml b/sample.yaml new file mode 100644 index 0000000..1a3241d --- /dev/null +++ b/sample.yaml @@ -0,0 +1,52 @@ +# 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