CreateInferenceExperimentRequest.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMakerRequest.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/InferenceExperimentDataStorageConfig.h>
#include <aws/sagemaker/model/InferenceExperimentSchedule.h>
#include <aws/sagemaker/model/InferenceExperimentType.h>
#include <aws/sagemaker/model/ModelVariantConfig.h>
#include <aws/sagemaker/model/ShadowModeConfig.h>
#include <aws/sagemaker/model/Tag.h>
 
#include <utility>
 
namespace Aws {
namespace SageMaker {
namespace Model {
 
class CreateInferenceExperimentRequest : public SageMakerRequest {
 public:
  AWS_SAGEMAKER_API CreateInferenceExperimentRequest() = default;
 
  // Service request name is the Operation name which will send this request out,
  // each operation should has unique request name, so that we can get operation's name from this request.
  // Note: this is not true for response, multiple operations may have the same response name,
  // so we can not get operation's name from response.
  inline virtual const char* GetServiceRequestName() const override { return "CreateInferenceExperiment"; }
 
  AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
 
  AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetName() const { return m_name; }
  inline bool NameHasBeenSet() const { return m_nameHasBeenSet; }
  template <typename NameT = Aws::String>
  void SetName(NameT&& value) {
    m_nameHasBeenSet = true;
    m_name = std::forward<NameT>(value);
  }
  template <typename NameT = Aws::String>
  CreateInferenceExperimentRequest& WithName(NameT&& value) {
    SetName(std::forward<NameT>(value));
    return *this;
  }
 
 
  inline InferenceExperimentType GetType() const { return m_type; }
  inline bool TypeHasBeenSet() const { return m_typeHasBeenSet; }
  inline void SetType(InferenceExperimentType value) {
    m_typeHasBeenSet = true;
    m_type = value;
  }
  inline CreateInferenceExperimentRequest& WithType(InferenceExperimentType value) {
    SetType(value);
    return *this;
  }
 
 
  inline const InferenceExperimentSchedule& GetSchedule() const { return m_schedule; }
  inline bool ScheduleHasBeenSet() const { return m_scheduleHasBeenSet; }
  template <typename ScheduleT = InferenceExperimentSchedule>
  void SetSchedule(ScheduleT&& value) {
    m_scheduleHasBeenSet = true;
    m_schedule = std::forward<ScheduleT>(value);
  }
  template <typename ScheduleT = InferenceExperimentSchedule>
  CreateInferenceExperimentRequest& WithSchedule(ScheduleT&& value) {
    SetSchedule(std::forward<ScheduleT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetDescription() const { return m_description; }
  inline bool DescriptionHasBeenSet() const { return m_descriptionHasBeenSet; }
  template <typename DescriptionT = Aws::String>
  void SetDescription(DescriptionT&& value) {
    m_descriptionHasBeenSet = true;
    m_description = std::forward<DescriptionT>(value);
  }
  template <typename DescriptionT = Aws::String>
  CreateInferenceExperimentRequest& WithDescription(DescriptionT&& value) {
    SetDescription(std::forward<DescriptionT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetRoleArn() const { return m_roleArn; }
  inline bool RoleArnHasBeenSet() const { return m_roleArnHasBeenSet; }
  template <typename RoleArnT = Aws::String>
  void SetRoleArn(RoleArnT&& value) {
    m_roleArnHasBeenSet = true;
    m_roleArn = std::forward<RoleArnT>(value);
  }
  template <typename RoleArnT = Aws::String>
  CreateInferenceExperimentRequest& WithRoleArn(RoleArnT&& value) {
    SetRoleArn(std::forward<RoleArnT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetEndpointName() const { return m_endpointName; }
  inline bool EndpointNameHasBeenSet() const { return m_endpointNameHasBeenSet; }
  template <typename EndpointNameT = Aws::String>
  void SetEndpointName(EndpointNameT&& value) {
    m_endpointNameHasBeenSet = true;
    m_endpointName = std::forward<EndpointNameT>(value);
  }
  template <typename EndpointNameT = Aws::String>
  CreateInferenceExperimentRequest& WithEndpointName(EndpointNameT&& value) {
    SetEndpointName(std::forward<EndpointNameT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<ModelVariantConfig>& GetModelVariants() const { return m_modelVariants; }
  inline bool ModelVariantsHasBeenSet() const { return m_modelVariantsHasBeenSet; }
  template <typename ModelVariantsT = Aws::Vector<ModelVariantConfig>>
  void SetModelVariants(ModelVariantsT&& value) {
    m_modelVariantsHasBeenSet = true;
    m_modelVariants = std::forward<ModelVariantsT>(value);
  }
  template <typename ModelVariantsT = Aws::Vector<ModelVariantConfig>>
  CreateInferenceExperimentRequest& WithModelVariants(ModelVariantsT&& value) {
    SetModelVariants(std::forward<ModelVariantsT>(value));
    return *this;
  }
  template <typename ModelVariantsT = ModelVariantConfig>
  CreateInferenceExperimentRequest& AddModelVariants(ModelVariantsT&& value) {
    m_modelVariantsHasBeenSet = true;
    m_modelVariants.emplace_back(std::forward<ModelVariantsT>(value));
    return *this;
  }
 
 
  inline const InferenceExperimentDataStorageConfig& GetDataStorageConfig() const { return m_dataStorageConfig; }
  inline bool DataStorageConfigHasBeenSet() const { return m_dataStorageConfigHasBeenSet; }
  template <typename DataStorageConfigT = InferenceExperimentDataStorageConfig>
  void SetDataStorageConfig(DataStorageConfigT&& value) {
    m_dataStorageConfigHasBeenSet = true;
    m_dataStorageConfig = std::forward<DataStorageConfigT>(value);
  }
  template <typename DataStorageConfigT = InferenceExperimentDataStorageConfig>
  CreateInferenceExperimentRequest& WithDataStorageConfig(DataStorageConfigT&& value) {
    SetDataStorageConfig(std::forward<DataStorageConfigT>(value));
    return *this;
  }
 
 
  inline const ShadowModeConfig& GetShadowModeConfig() const { return m_shadowModeConfig; }
  inline bool ShadowModeConfigHasBeenSet() const { return m_shadowModeConfigHasBeenSet; }
  template <typename ShadowModeConfigT = ShadowModeConfig>
  void SetShadowModeConfig(ShadowModeConfigT&& value) {
    m_shadowModeConfigHasBeenSet = true;
    m_shadowModeConfig = std::forward<ShadowModeConfigT>(value);
  }
  template <typename ShadowModeConfigT = ShadowModeConfig>
  CreateInferenceExperimentRequest& WithShadowModeConfig(ShadowModeConfigT&& value) {
    SetShadowModeConfig(std::forward<ShadowModeConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetKmsKey() const { return m_kmsKey; }
  inline bool KmsKeyHasBeenSet() const { return m_kmsKeyHasBeenSet; }
  template <typename KmsKeyT = Aws::String>
  void SetKmsKey(KmsKeyT&& value) {
    m_kmsKeyHasBeenSet = true;
    m_kmsKey = std::forward<KmsKeyT>(value);
  }
  template <typename KmsKeyT = Aws::String>
  CreateInferenceExperimentRequest& WithKmsKey(KmsKeyT&& value) {
    SetKmsKey(std::forward<KmsKeyT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Tag>& GetTags() const { return m_tags; }
  inline bool TagsHasBeenSet() const { return m_tagsHasBeenSet; }
  template <typename TagsT = Aws::Vector<Tag>>
  void SetTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags = std::forward<TagsT>(value);
  }
  template <typename TagsT = Aws::Vector<Tag>>
  CreateInferenceExperimentRequest& WithTags(TagsT&& value) {
    SetTags(std::forward<TagsT>(value));
    return *this;
  }
  template <typename TagsT = Tag>
  CreateInferenceExperimentRequest& AddTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags.emplace_back(std::forward<TagsT>(value));
    return *this;
  }
 private:
  Aws::String m_name;
 
  InferenceExperimentType m_type{InferenceExperimentType::NOT_SET};
 
  InferenceExperimentSchedule m_schedule;
 
  Aws::String m_description;
 
  Aws::String m_roleArn;
 
  Aws::String m_endpointName;
 
  Aws::Vector<ModelVariantConfig> m_modelVariants;
 
  InferenceExperimentDataStorageConfig m_dataStorageConfig;
 
  ShadowModeConfig m_shadowModeConfig;
 
  Aws::String m_kmsKey;
 
  Aws::Vector<Tag> m_tags;
  bool m_nameHasBeenSet = false;
  bool m_typeHasBeenSet = false;
  bool m_scheduleHasBeenSet = false;
  bool m_descriptionHasBeenSet = false;
  bool m_roleArnHasBeenSet = false;
  bool m_endpointNameHasBeenSet = false;
  bool m_modelVariantsHasBeenSet = false;
  bool m_dataStorageConfigHasBeenSet = false;
  bool m_shadowModeConfigHasBeenSet = false;
  bool m_kmsKeyHasBeenSet = false;
  bool m_tagsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws