name: "Sagemaker - ModelExplainabilityJobDefinition"
description: Create ModelExplainabilityJobDefinition
inputs:
  - {
      name: region,
      type: String,
      description: "The region to use for the training job",
    }
  ###########################GENERATED SECTION BELOW############################
  
  - {
      name: job_definition_name,
      type: String,
      default: '',
      description: "The name of the model explainability job definition.",
    }
  - {
      name: job_resources,
      type: JsonObject,
      default: '{}',
      description: "Identifies the resources to deploy for a monitoring job.",
    }
  - {
      name: model_explainability_app_specification,
      type: JsonObject,
      default: '{}',
      description: "Configures the model explainability job to run a specified Docker container image.",
    }
  - {
      name: model_explainability_baseline_config,
      type: JsonObject,
      default: '{}',
      description: "The baseline configuration for a model explainability job.",
    }
  - {
      name: model_explainability_job_input,
      type: JsonObject,
      default: '{}',
      description: "Inputs for the model explainability job.",
    }
  - {
      name: model_explainability_job_output_config,
      type: JsonObject,
      default: '{}',
      description: "The output configuration for monitoring jobs.",
    }
  - {
      name: network_config,
      type: JsonObject,
      default: '{}',
      description: "Networking options for a model explainability job.",
    }
  - {
      name: role_arn,
      type: String,
      default: '',
      description: "The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker can assume to perform tasks on your behalf.",
    }
  - {
      name: stopping_condition,
      type: JsonObject,
      default: '{}',
      description: "A time limit for how long the monitoring job is allowed to run before stopping.",
    }
  - {
      name: tags,
      type: JsonArray,
      default: '[]',
      description: "(Optional) An array of key-value pairs.",
    }
  ###########################GENERATED SECTION ABOVE############################

outputs:
  ###########################GENERATED SECTION BELOW############################
  
  - {
      name: ack_resource_metadata,
      type: JsonObject,
      description: "All CRs managed by ACK have a common `Status.",
    }
  - {
      name: conditions,
      type: JsonArray,
      description: "All CRS managed by ACK have a common `Status.",
    }
  - {
      name: sagemaker_resource_name,
      type: String,
      description: "Resource name on Sagemaker",
    }
  ###########################GENERATED SECTION ABOVE############################

implementation:
  container:
    image: public.ecr.aws/kubeflow-on-aws/aws-sagemaker-kfp-components:2.3.0
    command: [python3]
    args:
      - ModelExplainabilityJobDefinition/src/ModelExplainabilityJobDefinition_component.py
      - --region
      - { inputValue: region }
      ###########################GENERATED SECTION BELOW############################
      - --job_definition_name
      - { inputValue: job_definition_name }
      - --job_resources
      - { inputValue: job_resources }
      - --model_explainability_app_specification
      - { inputValue: model_explainability_app_specification }
      - --model_explainability_baseline_config
      - { inputValue: model_explainability_baseline_config }
      - --model_explainability_job_input
      - { inputValue: model_explainability_job_input }
      - --model_explainability_job_output_config
      - { inputValue: model_explainability_job_output_config }
      - --network_config
      - { inputValue: network_config }
      - --role_arn
      - { inputValue: role_arn }
      - --stopping_condition
      - { inputValue: stopping_condition }
      - --tags
      - { inputValue: tags }
      
      ###########################GENERATED SECTION ABOVE############################