{
  "train_component.yaml": {
    "implementation.container.command": ["python3", "bert/agnews_classification_pytorch.py"],
    "implementation.container.image": "public.ecr.aws/pytorch-samples/kfp_samples:latest"
  },
  "preprocess_component.yaml": {
    "implementation.container.command": ["python3", "bert/bert_pre_process.py"],
    "implementation.container.image": "public.ecr.aws/pytorch-samples/kfp_samples:latest"
  }
}

