name: Train XGBoost
description: Train an XGBoost
inputs:
  - {name: data_artifact, type: Dataset, optional: True, description: 'Dataset used for training'}
  - {name: data_path, type: String, description: 'Path to parquet file used for training'}
  - {name: target_variable, type: String, description: ''}
  
outputs:
  - {name: model_prod_artifact, type: Model}
  
implementation:
  container:
    image: gcr.io/PROJECT_ID/component_default@DIGEST
    args:
    - --executor_input
    - executorInput: null
    - --function_to_execute
    - train_xgb
    command: 
    - python3
    - -m 
    - kfp.dsl.executor_main
    - --component_module_path
    - ./train_xgb.py