RecommendationJobContainerConfig.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/RecommendationJobPayloadConfig.h>
#include <aws/sagemaker/model/RecommendationJobSupportedEndpointType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class RecommendationJobContainerConfig {
 public:
  AWS_SAGEMAKER_API RecommendationJobContainerConfig() = default;
  AWS_SAGEMAKER_API RecommendationJobContainerConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API RecommendationJobContainerConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::String& GetDomain() const { return m_domain; }
  inline bool DomainHasBeenSet() const { return m_domainHasBeenSet; }
  template <typename DomainT = Aws::String>
  void SetDomain(DomainT&& value) {
    m_domainHasBeenSet = true;
    m_domain = std::forward<DomainT>(value);
  }
  template <typename DomainT = Aws::String>
  RecommendationJobContainerConfig& WithDomain(DomainT&& value) {
    SetDomain(std::forward<DomainT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTask() const { return m_task; }
  inline bool TaskHasBeenSet() const { return m_taskHasBeenSet; }
  template <typename TaskT = Aws::String>
  void SetTask(TaskT&& value) {
    m_taskHasBeenSet = true;
    m_task = std::forward<TaskT>(value);
  }
  template <typename TaskT = Aws::String>
  RecommendationJobContainerConfig& WithTask(TaskT&& value) {
    SetTask(std::forward<TaskT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetFramework() const { return m_framework; }
  inline bool FrameworkHasBeenSet() const { return m_frameworkHasBeenSet; }
  template <typename FrameworkT = Aws::String>
  void SetFramework(FrameworkT&& value) {
    m_frameworkHasBeenSet = true;
    m_framework = std::forward<FrameworkT>(value);
  }
  template <typename FrameworkT = Aws::String>
  RecommendationJobContainerConfig& WithFramework(FrameworkT&& value) {
    SetFramework(std::forward<FrameworkT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetFrameworkVersion() const { return m_frameworkVersion; }
  inline bool FrameworkVersionHasBeenSet() const { return m_frameworkVersionHasBeenSet; }
  template <typename FrameworkVersionT = Aws::String>
  void SetFrameworkVersion(FrameworkVersionT&& value) {
    m_frameworkVersionHasBeenSet = true;
    m_frameworkVersion = std::forward<FrameworkVersionT>(value);
  }
  template <typename FrameworkVersionT = Aws::String>
  RecommendationJobContainerConfig& WithFrameworkVersion(FrameworkVersionT&& value) {
    SetFrameworkVersion(std::forward<FrameworkVersionT>(value));
    return *this;
  }
 
 
  inline const RecommendationJobPayloadConfig& GetPayloadConfig() const { return m_payloadConfig; }
  inline bool PayloadConfigHasBeenSet() const { return m_payloadConfigHasBeenSet; }
  template <typename PayloadConfigT = RecommendationJobPayloadConfig>
  void SetPayloadConfig(PayloadConfigT&& value) {
    m_payloadConfigHasBeenSet = true;
    m_payloadConfig = std::forward<PayloadConfigT>(value);
  }
  template <typename PayloadConfigT = RecommendationJobPayloadConfig>
  RecommendationJobContainerConfig& WithPayloadConfig(PayloadConfigT&& value) {
    SetPayloadConfig(std::forward<PayloadConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetNearestModelName() const { return m_nearestModelName; }
  inline bool NearestModelNameHasBeenSet() const { return m_nearestModelNameHasBeenSet; }
  template <typename NearestModelNameT = Aws::String>
  void SetNearestModelName(NearestModelNameT&& value) {
    m_nearestModelNameHasBeenSet = true;
    m_nearestModelName = std::forward<NearestModelNameT>(value);
  }
  template <typename NearestModelNameT = Aws::String>
  RecommendationJobContainerConfig& WithNearestModelName(NearestModelNameT&& value) {
    SetNearestModelName(std::forward<NearestModelNameT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetSupportedInstanceTypes() const { return m_supportedInstanceTypes; }
  inline bool SupportedInstanceTypesHasBeenSet() const { return m_supportedInstanceTypesHasBeenSet; }
  template <typename SupportedInstanceTypesT = Aws::Vector<Aws::String>>
  void SetSupportedInstanceTypes(SupportedInstanceTypesT&& value) {
    m_supportedInstanceTypesHasBeenSet = true;
    m_supportedInstanceTypes = std::forward<SupportedInstanceTypesT>(value);
  }
  template <typename SupportedInstanceTypesT = Aws::Vector<Aws::String>>
  RecommendationJobContainerConfig& WithSupportedInstanceTypes(SupportedInstanceTypesT&& value) {
    SetSupportedInstanceTypes(std::forward<SupportedInstanceTypesT>(value));
    return *this;
  }
  template <typename SupportedInstanceTypesT = Aws::String>
  RecommendationJobContainerConfig& AddSupportedInstanceTypes(SupportedInstanceTypesT&& value) {
    m_supportedInstanceTypesHasBeenSet = true;
    m_supportedInstanceTypes.emplace_back(std::forward<SupportedInstanceTypesT>(value));
    return *this;
  }
 
 
  inline RecommendationJobSupportedEndpointType GetSupportedEndpointType() const { return m_supportedEndpointType; }
  inline bool SupportedEndpointTypeHasBeenSet() const { return m_supportedEndpointTypeHasBeenSet; }
  inline void SetSupportedEndpointType(RecommendationJobSupportedEndpointType value) {
    m_supportedEndpointTypeHasBeenSet = true;
    m_supportedEndpointType = value;
  }
  inline RecommendationJobContainerConfig& WithSupportedEndpointType(RecommendationJobSupportedEndpointType value) {
    SetSupportedEndpointType(value);
    return *this;
  }
 
 
  inline const Aws::String& GetDataInputConfig() const { return m_dataInputConfig; }
  inline bool DataInputConfigHasBeenSet() const { return m_dataInputConfigHasBeenSet; }
  template <typename DataInputConfigT = Aws::String>
  void SetDataInputConfig(DataInputConfigT&& value) {
    m_dataInputConfigHasBeenSet = true;
    m_dataInputConfig = std::forward<DataInputConfigT>(value);
  }
  template <typename DataInputConfigT = Aws::String>
  RecommendationJobContainerConfig& WithDataInputConfig(DataInputConfigT&& value) {
    SetDataInputConfig(std::forward<DataInputConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetSupportedResponseMIMETypes() const { return m_supportedResponseMIMETypes; }
  inline bool SupportedResponseMIMETypesHasBeenSet() const { return m_supportedResponseMIMETypesHasBeenSet; }
  template <typename SupportedResponseMIMETypesT = Aws::Vector<Aws::String>>
  void SetSupportedResponseMIMETypes(SupportedResponseMIMETypesT&& value) {
    m_supportedResponseMIMETypesHasBeenSet = true;
    m_supportedResponseMIMETypes = std::forward<SupportedResponseMIMETypesT>(value);
  }
  template <typename SupportedResponseMIMETypesT = Aws::Vector<Aws::String>>
  RecommendationJobContainerConfig& WithSupportedResponseMIMETypes(SupportedResponseMIMETypesT&& value) {
    SetSupportedResponseMIMETypes(std::forward<SupportedResponseMIMETypesT>(value));
    return *this;
  }
  template <typename SupportedResponseMIMETypesT = Aws::String>
  RecommendationJobContainerConfig& AddSupportedResponseMIMETypes(SupportedResponseMIMETypesT&& value) {
    m_supportedResponseMIMETypesHasBeenSet = true;
    m_supportedResponseMIMETypes.emplace_back(std::forward<SupportedResponseMIMETypesT>(value));
    return *this;
  }
 private:
  Aws::String m_domain;
 
  Aws::String m_task;
 
  Aws::String m_framework;
 
  Aws::String m_frameworkVersion;
 
  RecommendationJobPayloadConfig m_payloadConfig;
 
  Aws::String m_nearestModelName;
 
  Aws::Vector<Aws::String> m_supportedInstanceTypes;
 
  RecommendationJobSupportedEndpointType m_supportedEndpointType{RecommendationJobSupportedEndpointType::NOT_SET};
 
  Aws::String m_dataInputConfig;
 
  Aws::Vector<Aws::String> m_supportedResponseMIMETypes;
  bool m_domainHasBeenSet = false;
  bool m_taskHasBeenSet = false;
  bool m_frameworkHasBeenSet = false;
  bool m_frameworkVersionHasBeenSet = false;
  bool m_payloadConfigHasBeenSet = false;
  bool m_nearestModelNameHasBeenSet = false;
  bool m_supportedInstanceTypesHasBeenSet = false;
  bool m_supportedEndpointTypeHasBeenSet = false;
  bool m_dataInputConfigHasBeenSet = false;
  bool m_supportedResponseMIMETypesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws