InferenceSpecification.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/ModelPackageContainerDefinition.h>
#include <aws/sagemaker/model/ProductionVariantInstanceType.h>
#include <aws/sagemaker/model/TransformInstanceType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class InferenceSpecification {
 public:
  AWS_SAGEMAKER_API InferenceSpecification() = default;
  AWS_SAGEMAKER_API InferenceSpecification(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API InferenceSpecification& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::Vector<ModelPackageContainerDefinition>& GetContainers() const { return m_containers; }
  inline bool ContainersHasBeenSet() const { return m_containersHasBeenSet; }
  template <typename ContainersT = Aws::Vector<ModelPackageContainerDefinition>>
  void SetContainers(ContainersT&& value) {
    m_containersHasBeenSet = true;
    m_containers = std::forward<ContainersT>(value);
  }
  template <typename ContainersT = Aws::Vector<ModelPackageContainerDefinition>>
  InferenceSpecification& WithContainers(ContainersT&& value) {
    SetContainers(std::forward<ContainersT>(value));
    return *this;
  }
  template <typename ContainersT = ModelPackageContainerDefinition>
  InferenceSpecification& AddContainers(ContainersT&& value) {
    m_containersHasBeenSet = true;
    m_containers.emplace_back(std::forward<ContainersT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<TransformInstanceType>& GetSupportedTransformInstanceTypes() const { return m_supportedTransformInstanceTypes; }
  inline bool SupportedTransformInstanceTypesHasBeenSet() const { return m_supportedTransformInstanceTypesHasBeenSet; }
  template <typename SupportedTransformInstanceTypesT = Aws::Vector<TransformInstanceType>>
  void SetSupportedTransformInstanceTypes(SupportedTransformInstanceTypesT&& value) {
    m_supportedTransformInstanceTypesHasBeenSet = true;
    m_supportedTransformInstanceTypes = std::forward<SupportedTransformInstanceTypesT>(value);
  }
  template <typename SupportedTransformInstanceTypesT = Aws::Vector<TransformInstanceType>>
  InferenceSpecification& WithSupportedTransformInstanceTypes(SupportedTransformInstanceTypesT&& value) {
    SetSupportedTransformInstanceTypes(std::forward<SupportedTransformInstanceTypesT>(value));
    return *this;
  }
  inline InferenceSpecification& AddSupportedTransformInstanceTypes(TransformInstanceType value) {
    m_supportedTransformInstanceTypesHasBeenSet = true;
    m_supportedTransformInstanceTypes.push_back(value);
    return *this;
  }
 
 
  inline const Aws::Vector<ProductionVariantInstanceType>& GetSupportedRealtimeInferenceInstanceTypes() const {
    return m_supportedRealtimeInferenceInstanceTypes;
  }
  inline bool SupportedRealtimeInferenceInstanceTypesHasBeenSet() const { return m_supportedRealtimeInferenceInstanceTypesHasBeenSet; }
  template <typename SupportedRealtimeInferenceInstanceTypesT = Aws::Vector<ProductionVariantInstanceType>>
  void SetSupportedRealtimeInferenceInstanceTypes(SupportedRealtimeInferenceInstanceTypesT&& value) {
    m_supportedRealtimeInferenceInstanceTypesHasBeenSet = true;
    m_supportedRealtimeInferenceInstanceTypes = std::forward<SupportedRealtimeInferenceInstanceTypesT>(value);
  }
  template <typename SupportedRealtimeInferenceInstanceTypesT = Aws::Vector<ProductionVariantInstanceType>>
  InferenceSpecification& WithSupportedRealtimeInferenceInstanceTypes(SupportedRealtimeInferenceInstanceTypesT&& value) {
    SetSupportedRealtimeInferenceInstanceTypes(std::forward<SupportedRealtimeInferenceInstanceTypesT>(value));
    return *this;
  }
  inline InferenceSpecification& AddSupportedRealtimeInferenceInstanceTypes(ProductionVariantInstanceType value) {
    m_supportedRealtimeInferenceInstanceTypesHasBeenSet = true;
    m_supportedRealtimeInferenceInstanceTypes.push_back(value);
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetSupportedContentTypes() const { return m_supportedContentTypes; }
  inline bool SupportedContentTypesHasBeenSet() const { return m_supportedContentTypesHasBeenSet; }
  template <typename SupportedContentTypesT = Aws::Vector<Aws::String>>
  void SetSupportedContentTypes(SupportedContentTypesT&& value) {
    m_supportedContentTypesHasBeenSet = true;
    m_supportedContentTypes = std::forward<SupportedContentTypesT>(value);
  }
  template <typename SupportedContentTypesT = Aws::Vector<Aws::String>>
  InferenceSpecification& WithSupportedContentTypes(SupportedContentTypesT&& value) {
    SetSupportedContentTypes(std::forward<SupportedContentTypesT>(value));
    return *this;
  }
  template <typename SupportedContentTypesT = Aws::String>
  InferenceSpecification& AddSupportedContentTypes(SupportedContentTypesT&& value) {
    m_supportedContentTypesHasBeenSet = true;
    m_supportedContentTypes.emplace_back(std::forward<SupportedContentTypesT>(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>>
  InferenceSpecification& WithSupportedResponseMIMETypes(SupportedResponseMIMETypesT&& value) {
    SetSupportedResponseMIMETypes(std::forward<SupportedResponseMIMETypesT>(value));
    return *this;
  }
  template <typename SupportedResponseMIMETypesT = Aws::String>
  InferenceSpecification& AddSupportedResponseMIMETypes(SupportedResponseMIMETypesT&& value) {
    m_supportedResponseMIMETypesHasBeenSet = true;
    m_supportedResponseMIMETypes.emplace_back(std::forward<SupportedResponseMIMETypesT>(value));
    return *this;
  }
 private:
  Aws::Vector<ModelPackageContainerDefinition> m_containers;
 
  Aws::Vector<TransformInstanceType> m_supportedTransformInstanceTypes;
 
  Aws::Vector<ProductionVariantInstanceType> m_supportedRealtimeInferenceInstanceTypes;
 
  Aws::Vector<Aws::String> m_supportedContentTypes;
 
  Aws::Vector<Aws::String> m_supportedResponseMIMETypes;
  bool m_containersHasBeenSet = false;
  bool m_supportedTransformInstanceTypesHasBeenSet = false;
  bool m_supportedRealtimeInferenceInstanceTypesHasBeenSet = false;
  bool m_supportedContentTypesHasBeenSet = false;
  bool m_supportedResponseMIMETypesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws