From 73f4d1bd298ca5d46a96d176faeae10d56ed21fc Mon Sep 17 00:00:00 2001 From: Chang-Hong Hsu Date: Thu, 30 Jul 2020 10:52:33 -0700 Subject: [PATCH] Adding proto definitions for supporting SageMaker TrainingJob (built-in algorithms) and HyperparameterTuningJob (#66) * adding sagemaker protos Co-authored-by: Haytham AbuelFutuh --- .../hyperparameter_tuning_job.grpc.pb.cc | 26 + .../hyperparameter_tuning_job.grpc.pb.h | 49 + .../sagemaker/hyperparameter_tuning_job.pb.cc | 2174 +++++++ .../sagemaker/hyperparameter_tuning_job.pb.h | 1283 ++++ .../sagemaker/parameter_ranges.grpc.pb.cc | 26 + .../sagemaker/parameter_ranges.grpc.pb.h | 49 + .../plugins/sagemaker/parameter_ranges.pb.cc | 2460 ++++++++ .../plugins/sagemaker/parameter_ranges.pb.h | 1308 ++++ .../plugins/sagemaker/training_job.grpc.pb.cc | 26 + .../plugins/sagemaker/training_job.grpc.pb.h | 49 + .../plugins/sagemaker/training_job.pb.cc | 2628 ++++++++ .../plugins/sagemaker/training_job.pb.h | 1592 +++++ .../flyteidl/core/compiler.pb.validate.go | 34 +- .../flyteidl/core/interface.pb.validate.go | 34 +- .../flyteidl/core/literals.pb.validate.go | 34 +- .../sagemaker/hyperparameter_tuning_job.pb.go | 437 ++ .../hyperparameter_tuning_job.pb.validate.go | 486 ++ .../plugins/sagemaker/parameter_ranges.pb.go | 435 ++ .../sagemaker/parameter_ranges.pb.validate.go | 506 ++ .../plugins/sagemaker/training_job.pb.go | 509 ++ .../sagemaker/training_job.pb.validate.go | 548 ++ gen/pb-go/flyteidl/service/openapi.go | 5 +- .../HyperparameterTuningJobOuterClass.java | 4496 +++++++++++++ .../sagemaker/ParameterRangesOuterClass.java | 4468 +++++++++++++ .../sagemaker/TrainingJobOuterClass.java | 5615 +++++++++++++++++ gen/pb-protodoc/flyteidl/plugins/index.rst | 1 + .../hyperparameter_tuning_job.proto.rst | 230 + .../flyteidl/plugins/sagemaker/index.rst | 11 + .../sagemaker/parameter_ranges.proto.rst | 215 + .../plugins/sagemaker/training_job.proto.rst | 274 + .../flyteidl/plugins/sagemaker/__init__.py | 0 .../hyperparameter_tuning_job_pb2.py | 368 ++ .../hyperparameter_tuning_job_pb2_grpc.py | 3 + .../plugins/sagemaker/parameter_ranges_pb2.py | 399 ++ .../sagemaker/parameter_ranges_pb2_grpc.py | 3 + .../plugins/sagemaker/training_job_pb2.py | 416 ++ .../sagemaker/training_job_pb2_grpc.py | 3 + generate_protos.sh | 2 +- package.json | 2 +- .../sagemaker/hyperparameter_tuning_job.proto | 83 + .../plugins/sagemaker/parameter_ranges.proto | 61 + .../plugins/sagemaker/training_job.proto | 100 + setup.py | 2 +- 43 files changed, 31439 insertions(+), 11 deletions(-) create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.h create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.h create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.h create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.h create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.cc create mode 100644 gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.h create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.go create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.validate.go create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.go create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.validate.go create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.go create mode 100644 gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.validate.go create mode 100644 gen/pb-java/flyteidl/plugins/sagemaker/HyperparameterTuningJobOuterClass.java create mode 100644 gen/pb-java/flyteidl/plugins/sagemaker/ParameterRangesOuterClass.java create mode 100644 gen/pb-java/flyteidl/plugins/sagemaker/TrainingJobOuterClass.java create mode 100644 gen/pb-protodoc/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto.rst create mode 100644 gen/pb-protodoc/flyteidl/plugins/sagemaker/index.rst create mode 100644 gen/pb-protodoc/flyteidl/plugins/sagemaker/parameter_ranges.proto.rst create mode 100644 gen/pb-protodoc/flyteidl/plugins/sagemaker/training_job.proto.rst create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/__init__.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2_grpc.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2_grpc.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2.py create mode 100644 gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2_grpc.py create mode 100644 protos/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto create mode 100644 protos/flyteidl/plugins/sagemaker/parameter_ranges.proto create mode 100644 protos/flyteidl/plugins/sagemaker/training_job.proto diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.cc new file mode 100644 index 000000000..a47db3745 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.cc @@ -0,0 +1,26 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +#include "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h" +#include "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace flyteidl +} // namespace plugins +} // namespace sagemaker + diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.h new file mode 100644 index 000000000..c01143b03 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.grpc.pb.h @@ -0,0 +1,49 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto +#ifndef GRPC_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto__INCLUDED +#define GRPC_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto__INCLUDED + +#include "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +namespace grpc_impl { +class Channel; +class CompletionQueue; +class ServerCompletionQueue; +} // namespace grpc_impl + +namespace grpc { +namespace experimental { +template +class MessageAllocator; +} // namespace experimental +} // namespace grpc_impl + +namespace grpc { +class ServerContext; +} // namespace grpc + +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + + +#endif // GRPC_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto__INCLUDED diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.cc new file mode 100644 index 000000000..a1a5f3252 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.cc @@ -0,0 +1,2174 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +#include "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h" + +#include + +#include +#include +#include +#include +#include +#include +#include +#include +// @@protoc_insertion_point(includes) +#include + +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<1> scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto ::google::protobuf::internal::SCCInfo<2> scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class HyperparameterTuningJobDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterTuningJob_default_instance_; +class HyperparameterTuningObjectiveTypeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterTuningObjectiveType_default_instance_; +class HyperparameterTuningObjectiveDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterTuningObjective_default_instance_; +class HyperparameterTuningStrategyDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterTuningStrategy_default_instance_; +class TrainingJobEarlyStoppingTypeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _TrainingJobEarlyStoppingType_default_instance_; +class HyperparameterTuningJobConfigDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterTuningJobConfig_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +static void InitDefaultsHyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterTuningJob_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterTuningJob(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterTuningJob::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<1> scc_info_HyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 1, InitDefaultsHyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, { + &scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base,}}; + +static void InitDefaultsHyperparameterTuningObjectiveType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterTuningObjectiveType_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_HyperparameterTuningObjectiveType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsHyperparameterTuningObjectiveType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, {}}; + +static void InitDefaultsHyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterTuningObjective_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsHyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, {}}; + +static void InitDefaultsHyperparameterTuningStrategy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterTuningStrategy_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_HyperparameterTuningStrategy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsHyperparameterTuningStrategy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, {}}; + +static void InitDefaultsTrainingJobEarlyStoppingType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_TrainingJobEarlyStoppingType_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_TrainingJobEarlyStoppingType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsTrainingJobEarlyStoppingType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, {}}; + +static void InitDefaultsHyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterTuningJobConfig_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<2> scc_info_HyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 2, InitDefaultsHyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto}, { + &scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base, + &scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base,}}; + +void InitDefaults_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterTuningObjectiveType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterTuningStrategy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_TrainingJobEarlyStoppingType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); +} + +::google::protobuf::Metadata file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[6]; +const ::google::protobuf::EnumDescriptor* file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[3]; +constexpr ::google::protobuf::ServiceDescriptor const** file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = nullptr; + +const ::google::protobuf::uint32 TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto::offsets[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJob, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJob, training_job_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJob, max_number_of_training_jobs_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJob, max_parallel_training_jobs_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective, objective_type_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective, metric_name_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig, hyperparameter_ranges_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig, tuning_strategy_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig, tuning_objective_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig, training_job_early_stopping_type_), +}; +static const ::google::protobuf::internal::MigrationSchema schemas[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + { 0, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterTuningJob)}, + { 8, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType)}, + { 13, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective)}, + { 20, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy)}, + { 25, -1, sizeof(::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType)}, + { 30, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig)}, +}; + +static ::google::protobuf::Message const * const file_default_instances[] = { + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterTuningJob_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterTuningObjectiveType_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterTuningObjective_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterTuningStrategy_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_TrainingJobEarlyStoppingType_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterTuningJobConfig_default_instance_), +}; + +::google::protobuf::internal::AssignDescriptorsTable assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = { + {}, AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto", schemas, + file_default_instances, TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto::offsets, + file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, 6, file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, +}; + +const char descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[] = + "\n:flyteidl/plugins/sagemaker/hyperparame" + "ter_tuning_job.proto\022\032flyteidl.plugins.s" + "agemaker\0321flyteidl/plugins/sagemaker/par" + "ameter_ranges.proto\032-flyteidl/plugins/sa" + "gemaker/training_job.proto\"\241\001\n\027Hyperpara" + "meterTuningJob\022=\n\014training_job\030\001 \001(\0132\'.f" + "lyteidl.plugins.sagemaker.TrainingJob\022#\n" + "\033max_number_of_training_jobs\030\002 \001(\003\022\"\n\032ma" + "x_parallel_training_jobs\030\003 \001(\003\"H\n!Hyperp" + "arameterTuningObjectiveType\"#\n\005Value\022\014\n\010" + "MINIMIZE\020\000\022\014\n\010MAXIMIZE\020\001\"\221\001\n\035Hyperparame" + "terTuningObjective\022[\n\016objective_type\030\001 \001" + "(\0162C.flyteidl.plugins.sagemaker.Hyperpar" + "ameterTuningObjectiveType.Value\022\023\n\013metri" + "c_name\030\002 \001(\t\"A\n\034HyperparameterTuningStra" + "tegy\"!\n\005Value\022\014\n\010BAYESIAN\020\000\022\n\n\006RANDOM\020\001\"" + ":\n\034TrainingJobEarlyStoppingType\"\032\n\005Value" + "\022\007\n\003OFF\020\000\022\010\n\004AUTO\020\001\"\203\003\n\035HyperparameterTu" + "ningJobConfig\022J\n\025hyperparameter_ranges\030\001" + " \001(\0132+.flyteidl.plugins.sagemaker.Parame" + "terRanges\022W\n\017tuning_strategy\030\002 \001(\0162>.fly" + "teidl.plugins.sagemaker.HyperparameterTu" + "ningStrategy.Value\022S\n\020tuning_objective\030\003" + " \001(\01329.flyteidl.plugins.sagemaker.Hyperp" + "arameterTuningObjective\022h\n training_job_" + "early_stopping_type\030\004 \001(\0162>.flyteidl.plu" + "gins.sagemaker.TrainingJobEarlyStoppingT" + "ype.ValueB5Z3github.com/lyft/flyteidl/ge" + "n/pb-go/flyteidl/pluginsb\006proto3" + ; +::google::protobuf::internal::DescriptorTable descriptor_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = { + false, InitDefaults_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, + descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, + "flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto", &assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, 1152, +}; + +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto() { + static constexpr ::google::protobuf::internal::InitFunc deps[2] = + { + ::AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, + ::AddDescriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, + }; + ::google::protobuf::internal::AddDescriptors(&descriptor_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto, deps, 2); +} + +// Force running AddDescriptors() at dynamic initialization time. +static bool dynamic_init_dummy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto = []() { AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto(); return true; }(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +const ::google::protobuf::EnumDescriptor* HyperparameterTuningObjectiveType_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[0]; +} +bool HyperparameterTuningObjectiveType_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType::MINIMIZE; +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType::MAXIMIZE; +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType::Value_MIN; +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType::Value_MAX; +const int HyperparameterTuningObjectiveType::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 +const ::google::protobuf::EnumDescriptor* HyperparameterTuningStrategy_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[1]; +} +bool HyperparameterTuningStrategy_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy::BAYESIAN; +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy::RANDOM; +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy::Value_MIN; +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy::Value_MAX; +const int HyperparameterTuningStrategy::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 +const ::google::protobuf::EnumDescriptor* TrainingJobEarlyStoppingType_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[2]; +} +bool TrainingJobEarlyStoppingType_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType::OFF; +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType::AUTO; +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType::Value_MIN; +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType::Value_MAX; +const int TrainingJobEarlyStoppingType::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +// =================================================================== + +void HyperparameterTuningJob::InitAsDefaultInstance() { + ::flyteidl::plugins::sagemaker::_HyperparameterTuningJob_default_instance_._instance.get_mutable()->training_job_ = const_cast< ::flyteidl::plugins::sagemaker::TrainingJob*>( + ::flyteidl::plugins::sagemaker::TrainingJob::internal_default_instance()); +} +class HyperparameterTuningJob::HasBitSetters { + public: + static const ::flyteidl::plugins::sagemaker::TrainingJob& training_job(const HyperparameterTuningJob* msg); +}; + +const ::flyteidl::plugins::sagemaker::TrainingJob& +HyperparameterTuningJob::HasBitSetters::training_job(const HyperparameterTuningJob* msg) { + return *msg->training_job_; +} +void HyperparameterTuningJob::clear_training_job() { + if (GetArenaNoVirtual() == nullptr && training_job_ != nullptr) { + delete training_job_; + } + training_job_ = nullptr; +} +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int HyperparameterTuningJob::kTrainingJobFieldNumber; +const int HyperparameterTuningJob::kMaxNumberOfTrainingJobsFieldNumber; +const int HyperparameterTuningJob::kMaxParallelTrainingJobsFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterTuningJob::HyperparameterTuningJob() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterTuningJob) +} +HyperparameterTuningJob::HyperparameterTuningJob(const HyperparameterTuningJob& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + if (from.has_training_job()) { + training_job_ = new ::flyteidl::plugins::sagemaker::TrainingJob(*from.training_job_); + } else { + training_job_ = nullptr; + } + ::memcpy(&max_number_of_training_jobs_, &from.max_number_of_training_jobs_, + static_cast(reinterpret_cast(&max_parallel_training_jobs_) - + reinterpret_cast(&max_number_of_training_jobs_)) + sizeof(max_parallel_training_jobs_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterTuningJob) +} + +void HyperparameterTuningJob::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_HyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::memset(&training_job_, 0, static_cast( + reinterpret_cast(&max_parallel_training_jobs_) - + reinterpret_cast(&training_job_)) + sizeof(max_parallel_training_jobs_)); +} + +HyperparameterTuningJob::~HyperparameterTuningJob() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + SharedDtor(); +} + +void HyperparameterTuningJob::SharedDtor() { + if (this != internal_default_instance()) delete training_job_; +} + +void HyperparameterTuningJob::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterTuningJob& HyperparameterTuningJob::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterTuningJob_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterTuningJob::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + if (GetArenaNoVirtual() == nullptr && training_job_ != nullptr) { + delete training_job_; + } + training_job_ = nullptr; + ::memset(&max_number_of_training_jobs_, 0, static_cast( + reinterpret_cast(&max_parallel_training_jobs_) - + reinterpret_cast(&max_number_of_training_jobs_)) + sizeof(max_parallel_training_jobs_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterTuningJob::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::TrainingJob::_InternalParse; + object = msg->mutable_training_job(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // int64 max_number_of_training_jobs = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 16) goto handle_unusual; + msg->set_max_number_of_training_jobs(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // int64 max_parallel_training_jobs = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 24) goto handle_unusual; + msg->set_max_parallel_training_jobs(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterTuningJob::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_training_job())); + } else { + goto handle_unusual; + } + break; + } + + // int64 max_number_of_training_jobs = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (16 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &max_number_of_training_jobs_))); + } else { + goto handle_unusual; + } + break; + } + + // int64 max_parallel_training_jobs = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (24 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &max_parallel_training_jobs_))); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterTuningJob::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + if (this->has_training_job()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 1, HasBitSetters::training_job(this), output); + } + + // int64 max_number_of_training_jobs = 2; + if (this->max_number_of_training_jobs() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(2, this->max_number_of_training_jobs(), output); + } + + // int64 max_parallel_training_jobs = 3; + if (this->max_parallel_training_jobs() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(3, this->max_parallel_training_jobs(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterTuningJob) +} + +::google::protobuf::uint8* HyperparameterTuningJob::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + if (this->has_training_job()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 1, HasBitSetters::training_job(this), target); + } + + // int64 max_number_of_training_jobs = 2; + if (this->max_number_of_training_jobs() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(2, this->max_number_of_training_jobs(), target); + } + + // int64 max_parallel_training_jobs = 3; + if (this->max_parallel_training_jobs() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(3, this->max_parallel_training_jobs(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + return target; +} + +size_t HyperparameterTuningJob::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + if (this->has_training_job()) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *training_job_); + } + + // int64 max_number_of_training_jobs = 2; + if (this->max_number_of_training_jobs() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->max_number_of_training_jobs()); + } + + // int64 max_parallel_training_jobs = 3; + if (this->max_parallel_training_jobs() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->max_parallel_training_jobs()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterTuningJob::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterTuningJob* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + MergeFrom(*source); + } +} + +void HyperparameterTuningJob::MergeFrom(const HyperparameterTuningJob& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.has_training_job()) { + mutable_training_job()->::flyteidl::plugins::sagemaker::TrainingJob::MergeFrom(from.training_job()); + } + if (from.max_number_of_training_jobs() != 0) { + set_max_number_of_training_jobs(from.max_number_of_training_jobs()); + } + if (from.max_parallel_training_jobs() != 0) { + set_max_parallel_training_jobs(from.max_parallel_training_jobs()); + } +} + +void HyperparameterTuningJob::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterTuningJob::CopyFrom(const HyperparameterTuningJob& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterTuningJob::IsInitialized() const { + return true; +} + +void HyperparameterTuningJob::Swap(HyperparameterTuningJob* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterTuningJob::InternalSwap(HyperparameterTuningJob* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(training_job_, other->training_job_); + swap(max_number_of_training_jobs_, other->max_number_of_training_jobs_); + swap(max_parallel_training_jobs_, other->max_parallel_training_jobs_); +} + +::google::protobuf::Metadata HyperparameterTuningJob::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void HyperparameterTuningObjectiveType::InitAsDefaultInstance() { +} +class HyperparameterTuningObjectiveType::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterTuningObjectiveType::HyperparameterTuningObjectiveType() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) +} +HyperparameterTuningObjectiveType::HyperparameterTuningObjectiveType(const HyperparameterTuningObjectiveType& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) +} + +void HyperparameterTuningObjectiveType::SharedCtor() { +} + +HyperparameterTuningObjectiveType::~HyperparameterTuningObjectiveType() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + SharedDtor(); +} + +void HyperparameterTuningObjectiveType::SharedDtor() { +} + +void HyperparameterTuningObjectiveType::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterTuningObjectiveType& HyperparameterTuningObjectiveType::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterTuningObjectiveType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterTuningObjectiveType::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterTuningObjectiveType::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterTuningObjectiveType::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterTuningObjectiveType::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) +} + +::google::protobuf::uint8* HyperparameterTuningObjectiveType::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + return target; +} + +size_t HyperparameterTuningObjectiveType::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterTuningObjectiveType::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterTuningObjectiveType* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + MergeFrom(*source); + } +} + +void HyperparameterTuningObjectiveType::MergeFrom(const HyperparameterTuningObjectiveType& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void HyperparameterTuningObjectiveType::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterTuningObjectiveType::CopyFrom(const HyperparameterTuningObjectiveType& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterTuningObjectiveType::IsInitialized() const { + return true; +} + +void HyperparameterTuningObjectiveType::Swap(HyperparameterTuningObjectiveType* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterTuningObjectiveType::InternalSwap(HyperparameterTuningObjectiveType* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata HyperparameterTuningObjectiveType::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void HyperparameterTuningObjective::InitAsDefaultInstance() { +} +class HyperparameterTuningObjective::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int HyperparameterTuningObjective::kObjectiveTypeFieldNumber; +const int HyperparameterTuningObjective::kMetricNameFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterTuningObjective::HyperparameterTuningObjective() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) +} +HyperparameterTuningObjective::HyperparameterTuningObjective(const HyperparameterTuningObjective& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + metric_name_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + if (from.metric_name().size() > 0) { + metric_name_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.metric_name_); + } + objective_type_ = from.objective_type_; + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) +} + +void HyperparameterTuningObjective::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + metric_name_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + objective_type_ = 0; +} + +HyperparameterTuningObjective::~HyperparameterTuningObjective() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + SharedDtor(); +} + +void HyperparameterTuningObjective::SharedDtor() { + metric_name_.DestroyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} + +void HyperparameterTuningObjective::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterTuningObjective& HyperparameterTuningObjective::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterTuningObjective_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterTuningObjective::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + metric_name_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + objective_type_ = 0; + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterTuningObjective::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 8) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_objective_type(static_cast<::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // string metric_name = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 18) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name"); + object = msg->mutable_metric_name(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +string_till_end: + static_cast<::std::string*>(object)->clear(); + static_cast<::std::string*>(object)->reserve(size); + goto len_delim_till_end; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterTuningObjective::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (8 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_objective_type(static_cast< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + // string metric_name = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (18 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->mutable_metric_name())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->metric_name().data(), static_cast(this->metric_name().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name")); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterTuningObjective::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + if (this->objective_type() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 1, this->objective_type(), output); + } + + // string metric_name = 2; + if (this->metric_name().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->metric_name().data(), static_cast(this->metric_name().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name"); + ::google::protobuf::internal::WireFormatLite::WriteStringMaybeAliased( + 2, this->metric_name(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) +} + +::google::protobuf::uint8* HyperparameterTuningObjective::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + if (this->objective_type() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 1, this->objective_type(), target); + } + + // string metric_name = 2; + if (this->metric_name().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->metric_name().data(), static_cast(this->metric_name().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name"); + target = + ::google::protobuf::internal::WireFormatLite::WriteStringToArray( + 2, this->metric_name(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + return target; +} + +size_t HyperparameterTuningObjective::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // string metric_name = 2; + if (this->metric_name().size() > 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::StringSize( + this->metric_name()); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + if (this->objective_type() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->objective_type()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterTuningObjective::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterTuningObjective* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + MergeFrom(*source); + } +} + +void HyperparameterTuningObjective::MergeFrom(const HyperparameterTuningObjective& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.metric_name().size() > 0) { + + metric_name_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.metric_name_); + } + if (from.objective_type() != 0) { + set_objective_type(from.objective_type()); + } +} + +void HyperparameterTuningObjective::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterTuningObjective::CopyFrom(const HyperparameterTuningObjective& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterTuningObjective::IsInitialized() const { + return true; +} + +void HyperparameterTuningObjective::Swap(HyperparameterTuningObjective* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterTuningObjective::InternalSwap(HyperparameterTuningObjective* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + metric_name_.Swap(&other->metric_name_, &::google::protobuf::internal::GetEmptyStringAlreadyInited(), + GetArenaNoVirtual()); + swap(objective_type_, other->objective_type_); +} + +::google::protobuf::Metadata HyperparameterTuningObjective::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void HyperparameterTuningStrategy::InitAsDefaultInstance() { +} +class HyperparameterTuningStrategy::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterTuningStrategy::HyperparameterTuningStrategy() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) +} +HyperparameterTuningStrategy::HyperparameterTuningStrategy(const HyperparameterTuningStrategy& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) +} + +void HyperparameterTuningStrategy::SharedCtor() { +} + +HyperparameterTuningStrategy::~HyperparameterTuningStrategy() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + SharedDtor(); +} + +void HyperparameterTuningStrategy::SharedDtor() { +} + +void HyperparameterTuningStrategy::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterTuningStrategy& HyperparameterTuningStrategy::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterTuningStrategy_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterTuningStrategy::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterTuningStrategy::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterTuningStrategy::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterTuningStrategy::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) +} + +::google::protobuf::uint8* HyperparameterTuningStrategy::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + return target; +} + +size_t HyperparameterTuningStrategy::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterTuningStrategy::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterTuningStrategy* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + MergeFrom(*source); + } +} + +void HyperparameterTuningStrategy::MergeFrom(const HyperparameterTuningStrategy& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void HyperparameterTuningStrategy::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterTuningStrategy::CopyFrom(const HyperparameterTuningStrategy& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterTuningStrategy::IsInitialized() const { + return true; +} + +void HyperparameterTuningStrategy::Swap(HyperparameterTuningStrategy* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterTuningStrategy::InternalSwap(HyperparameterTuningStrategy* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata HyperparameterTuningStrategy::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void TrainingJobEarlyStoppingType::InitAsDefaultInstance() { +} +class TrainingJobEarlyStoppingType::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +TrainingJobEarlyStoppingType::TrainingJobEarlyStoppingType() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) +} +TrainingJobEarlyStoppingType::TrainingJobEarlyStoppingType(const TrainingJobEarlyStoppingType& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) +} + +void TrainingJobEarlyStoppingType::SharedCtor() { +} + +TrainingJobEarlyStoppingType::~TrainingJobEarlyStoppingType() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + SharedDtor(); +} + +void TrainingJobEarlyStoppingType::SharedDtor() { +} + +void TrainingJobEarlyStoppingType::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const TrainingJobEarlyStoppingType& TrainingJobEarlyStoppingType::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_TrainingJobEarlyStoppingType_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void TrainingJobEarlyStoppingType::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* TrainingJobEarlyStoppingType::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool TrainingJobEarlyStoppingType::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void TrainingJobEarlyStoppingType::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) +} + +::google::protobuf::uint8* TrainingJobEarlyStoppingType::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + return target; +} + +size_t TrainingJobEarlyStoppingType::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void TrainingJobEarlyStoppingType::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + GOOGLE_DCHECK_NE(&from, this); + const TrainingJobEarlyStoppingType* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + MergeFrom(*source); + } +} + +void TrainingJobEarlyStoppingType::MergeFrom(const TrainingJobEarlyStoppingType& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void TrainingJobEarlyStoppingType::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void TrainingJobEarlyStoppingType::CopyFrom(const TrainingJobEarlyStoppingType& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool TrainingJobEarlyStoppingType::IsInitialized() const { + return true; +} + +void TrainingJobEarlyStoppingType::Swap(TrainingJobEarlyStoppingType* other) { + if (other == this) return; + InternalSwap(other); +} +void TrainingJobEarlyStoppingType::InternalSwap(TrainingJobEarlyStoppingType* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata TrainingJobEarlyStoppingType::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void HyperparameterTuningJobConfig::InitAsDefaultInstance() { + ::flyteidl::plugins::sagemaker::_HyperparameterTuningJobConfig_default_instance_._instance.get_mutable()->hyperparameter_ranges_ = const_cast< ::flyteidl::plugins::sagemaker::ParameterRanges*>( + ::flyteidl::plugins::sagemaker::ParameterRanges::internal_default_instance()); + ::flyteidl::plugins::sagemaker::_HyperparameterTuningJobConfig_default_instance_._instance.get_mutable()->tuning_objective_ = const_cast< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective*>( + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective::internal_default_instance()); +} +class HyperparameterTuningJobConfig::HasBitSetters { + public: + static const ::flyteidl::plugins::sagemaker::ParameterRanges& hyperparameter_ranges(const HyperparameterTuningJobConfig* msg); + static const ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective& tuning_objective(const HyperparameterTuningJobConfig* msg); +}; + +const ::flyteidl::plugins::sagemaker::ParameterRanges& +HyperparameterTuningJobConfig::HasBitSetters::hyperparameter_ranges(const HyperparameterTuningJobConfig* msg) { + return *msg->hyperparameter_ranges_; +} +const ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective& +HyperparameterTuningJobConfig::HasBitSetters::tuning_objective(const HyperparameterTuningJobConfig* msg) { + return *msg->tuning_objective_; +} +void HyperparameterTuningJobConfig::clear_hyperparameter_ranges() { + if (GetArenaNoVirtual() == nullptr && hyperparameter_ranges_ != nullptr) { + delete hyperparameter_ranges_; + } + hyperparameter_ranges_ = nullptr; +} +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int HyperparameterTuningJobConfig::kHyperparameterRangesFieldNumber; +const int HyperparameterTuningJobConfig::kTuningStrategyFieldNumber; +const int HyperparameterTuningJobConfig::kTuningObjectiveFieldNumber; +const int HyperparameterTuningJobConfig::kTrainingJobEarlyStoppingTypeFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterTuningJobConfig::HyperparameterTuningJobConfig() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) +} +HyperparameterTuningJobConfig::HyperparameterTuningJobConfig(const HyperparameterTuningJobConfig& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + if (from.has_hyperparameter_ranges()) { + hyperparameter_ranges_ = new ::flyteidl::plugins::sagemaker::ParameterRanges(*from.hyperparameter_ranges_); + } else { + hyperparameter_ranges_ = nullptr; + } + if (from.has_tuning_objective()) { + tuning_objective_ = new ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective(*from.tuning_objective_); + } else { + tuning_objective_ = nullptr; + } + ::memcpy(&tuning_strategy_, &from.tuning_strategy_, + static_cast(reinterpret_cast(&training_job_early_stopping_type_) - + reinterpret_cast(&tuning_strategy_)) + sizeof(training_job_early_stopping_type_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) +} + +void HyperparameterTuningJobConfig::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_HyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + ::memset(&hyperparameter_ranges_, 0, static_cast( + reinterpret_cast(&training_job_early_stopping_type_) - + reinterpret_cast(&hyperparameter_ranges_)) + sizeof(training_job_early_stopping_type_)); +} + +HyperparameterTuningJobConfig::~HyperparameterTuningJobConfig() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + SharedDtor(); +} + +void HyperparameterTuningJobConfig::SharedDtor() { + if (this != internal_default_instance()) delete hyperparameter_ranges_; + if (this != internal_default_instance()) delete tuning_objective_; +} + +void HyperparameterTuningJobConfig::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterTuningJobConfig& HyperparameterTuningJobConfig::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterTuningJobConfig_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterTuningJobConfig::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + if (GetArenaNoVirtual() == nullptr && hyperparameter_ranges_ != nullptr) { + delete hyperparameter_ranges_; + } + hyperparameter_ranges_ = nullptr; + if (GetArenaNoVirtual() == nullptr && tuning_objective_ != nullptr) { + delete tuning_objective_; + } + tuning_objective_ = nullptr; + ::memset(&tuning_strategy_, 0, static_cast( + reinterpret_cast(&training_job_early_stopping_type_) - + reinterpret_cast(&tuning_strategy_)) + sizeof(training_job_early_stopping_type_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterTuningJobConfig::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::ParameterRanges::_InternalParse; + object = msg->mutable_hyperparameter_ranges(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 16) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_tuning_strategy(static_cast<::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 26) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective::_InternalParse; + object = msg->mutable_tuning_objective(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + case 4: { + if (static_cast<::google::protobuf::uint8>(tag) != 32) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_training_job_early_stopping_type(static_cast<::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterTuningJobConfig::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_hyperparameter_ranges())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (16 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_tuning_strategy(static_cast< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (26 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_tuning_objective())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + case 4: { + if (static_cast< ::google::protobuf::uint8>(tag) == (32 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_training_job_early_stopping_type(static_cast< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterTuningJobConfig::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + if (this->has_hyperparameter_ranges()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 1, HasBitSetters::hyperparameter_ranges(this), output); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + if (this->tuning_strategy() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 2, this->tuning_strategy(), output); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + if (this->has_tuning_objective()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 3, HasBitSetters::tuning_objective(this), output); + } + + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + if (this->training_job_early_stopping_type() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 4, this->training_job_early_stopping_type(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) +} + +::google::protobuf::uint8* HyperparameterTuningJobConfig::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + if (this->has_hyperparameter_ranges()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 1, HasBitSetters::hyperparameter_ranges(this), target); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + if (this->tuning_strategy() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 2, this->tuning_strategy(), target); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + if (this->has_tuning_objective()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 3, HasBitSetters::tuning_objective(this), target); + } + + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + if (this->training_job_early_stopping_type() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 4, this->training_job_early_stopping_type(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + return target; +} + +size_t HyperparameterTuningJobConfig::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + if (this->has_hyperparameter_ranges()) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *hyperparameter_ranges_); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + if (this->has_tuning_objective()) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *tuning_objective_); + } + + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + if (this->tuning_strategy() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->tuning_strategy()); + } + + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + if (this->training_job_early_stopping_type() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->training_job_early_stopping_type()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterTuningJobConfig::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterTuningJobConfig* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + MergeFrom(*source); + } +} + +void HyperparameterTuningJobConfig::MergeFrom(const HyperparameterTuningJobConfig& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.has_hyperparameter_ranges()) { + mutable_hyperparameter_ranges()->::flyteidl::plugins::sagemaker::ParameterRanges::MergeFrom(from.hyperparameter_ranges()); + } + if (from.has_tuning_objective()) { + mutable_tuning_objective()->::flyteidl::plugins::sagemaker::HyperparameterTuningObjective::MergeFrom(from.tuning_objective()); + } + if (from.tuning_strategy() != 0) { + set_tuning_strategy(from.tuning_strategy()); + } + if (from.training_job_early_stopping_type() != 0) { + set_training_job_early_stopping_type(from.training_job_early_stopping_type()); + } +} + +void HyperparameterTuningJobConfig::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterTuningJobConfig::CopyFrom(const HyperparameterTuningJobConfig& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterTuningJobConfig::IsInitialized() const { + return true; +} + +void HyperparameterTuningJobConfig::Swap(HyperparameterTuningJobConfig* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterTuningJobConfig::InternalSwap(HyperparameterTuningJobConfig* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(hyperparameter_ranges_, other->hyperparameter_ranges_); + swap(tuning_objective_, other->tuning_objective_); + swap(tuning_strategy_, other->tuning_strategy_); + swap(training_job_early_stopping_type_, other->training_job_early_stopping_type_); +} + +::google::protobuf::Metadata HyperparameterTuningJobConfig::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto[kIndexInFileMessages]; +} + + +// @@protoc_insertion_point(namespace_scope) +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterTuningJob* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterTuningJob >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterTuningJob >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig >(arena); +} +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) +#include diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h new file mode 100644 index 000000000..1e09a9443 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.h @@ -0,0 +1,1283 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +#ifndef PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto +#define PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto + +#include +#include + +#include +#if PROTOBUF_VERSION < 3007000 +#error This file was generated by a newer version of protoc which is +#error incompatible with your Protocol Buffer headers. Please update +#error your headers. +#endif +#if 3007000 < PROTOBUF_MIN_PROTOC_VERSION +#error This file was generated by an older version of protoc which is +#error incompatible with your Protocol Buffer headers. Please +#error regenerate this file with a newer version of protoc. +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include // IWYU pragma: export +#include // IWYU pragma: export +#include +#include +#include "flyteidl/plugins/sagemaker/parameter_ranges.pb.h" +#include "flyteidl/plugins/sagemaker/training_job.pb.h" +// @@protoc_insertion_point(includes) +#include +#define PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto + +// Internal implementation detail -- do not use these members. +struct TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto { + static const ::google::protobuf::internal::ParseTableField entries[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::AuxillaryParseTableField aux[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::ParseTable schema[6] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::FieldMetadata field_metadata[]; + static const ::google::protobuf::internal::SerializationTable serialization_table[]; + static const ::google::protobuf::uint32 offsets[]; +}; +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class HyperparameterTuningJob; +class HyperparameterTuningJobDefaultTypeInternal; +extern HyperparameterTuningJobDefaultTypeInternal _HyperparameterTuningJob_default_instance_; +class HyperparameterTuningJobConfig; +class HyperparameterTuningJobConfigDefaultTypeInternal; +extern HyperparameterTuningJobConfigDefaultTypeInternal _HyperparameterTuningJobConfig_default_instance_; +class HyperparameterTuningObjective; +class HyperparameterTuningObjectiveDefaultTypeInternal; +extern HyperparameterTuningObjectiveDefaultTypeInternal _HyperparameterTuningObjective_default_instance_; +class HyperparameterTuningObjectiveType; +class HyperparameterTuningObjectiveTypeDefaultTypeInternal; +extern HyperparameterTuningObjectiveTypeDefaultTypeInternal _HyperparameterTuningObjectiveType_default_instance_; +class HyperparameterTuningStrategy; +class HyperparameterTuningStrategyDefaultTypeInternal; +extern HyperparameterTuningStrategyDefaultTypeInternal _HyperparameterTuningStrategy_default_instance_; +class TrainingJobEarlyStoppingType; +class TrainingJobEarlyStoppingTypeDefaultTypeInternal; +extern TrainingJobEarlyStoppingTypeDefaultTypeInternal _TrainingJobEarlyStoppingType_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> ::flyteidl::plugins::sagemaker::HyperparameterTuningJob* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningJob>(Arena*); +template<> ::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningJobConfig>(Arena*); +template<> ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningObjective>(Arena*); +template<> ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType>(Arena*); +template<> ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy>(Arena*); +template<> ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType>(Arena*); +} // namespace protobuf +} // namespace google +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +enum HyperparameterTuningObjectiveType_Value { + HyperparameterTuningObjectiveType_Value_MINIMIZE = 0, + HyperparameterTuningObjectiveType_Value_MAXIMIZE = 1, + HyperparameterTuningObjectiveType_Value_HyperparameterTuningObjectiveType_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + HyperparameterTuningObjectiveType_Value_HyperparameterTuningObjectiveType_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool HyperparameterTuningObjectiveType_Value_IsValid(int value); +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType_Value_Value_MIN = HyperparameterTuningObjectiveType_Value_MINIMIZE; +const HyperparameterTuningObjectiveType_Value HyperparameterTuningObjectiveType_Value_Value_MAX = HyperparameterTuningObjectiveType_Value_MAXIMIZE; +const int HyperparameterTuningObjectiveType_Value_Value_ARRAYSIZE = HyperparameterTuningObjectiveType_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* HyperparameterTuningObjectiveType_Value_descriptor(); +inline const ::std::string& HyperparameterTuningObjectiveType_Value_Name(HyperparameterTuningObjectiveType_Value value) { + return ::google::protobuf::internal::NameOfEnum( + HyperparameterTuningObjectiveType_Value_descriptor(), value); +} +inline bool HyperparameterTuningObjectiveType_Value_Parse( + const ::std::string& name, HyperparameterTuningObjectiveType_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + HyperparameterTuningObjectiveType_Value_descriptor(), name, value); +} +enum HyperparameterTuningStrategy_Value { + HyperparameterTuningStrategy_Value_BAYESIAN = 0, + HyperparameterTuningStrategy_Value_RANDOM = 1, + HyperparameterTuningStrategy_Value_HyperparameterTuningStrategy_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + HyperparameterTuningStrategy_Value_HyperparameterTuningStrategy_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool HyperparameterTuningStrategy_Value_IsValid(int value); +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy_Value_Value_MIN = HyperparameterTuningStrategy_Value_BAYESIAN; +const HyperparameterTuningStrategy_Value HyperparameterTuningStrategy_Value_Value_MAX = HyperparameterTuningStrategy_Value_RANDOM; +const int HyperparameterTuningStrategy_Value_Value_ARRAYSIZE = HyperparameterTuningStrategy_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* HyperparameterTuningStrategy_Value_descriptor(); +inline const ::std::string& HyperparameterTuningStrategy_Value_Name(HyperparameterTuningStrategy_Value value) { + return ::google::protobuf::internal::NameOfEnum( + HyperparameterTuningStrategy_Value_descriptor(), value); +} +inline bool HyperparameterTuningStrategy_Value_Parse( + const ::std::string& name, HyperparameterTuningStrategy_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + HyperparameterTuningStrategy_Value_descriptor(), name, value); +} +enum TrainingJobEarlyStoppingType_Value { + TrainingJobEarlyStoppingType_Value_OFF = 0, + TrainingJobEarlyStoppingType_Value_AUTO = 1, + TrainingJobEarlyStoppingType_Value_TrainingJobEarlyStoppingType_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + TrainingJobEarlyStoppingType_Value_TrainingJobEarlyStoppingType_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool TrainingJobEarlyStoppingType_Value_IsValid(int value); +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType_Value_Value_MIN = TrainingJobEarlyStoppingType_Value_OFF; +const TrainingJobEarlyStoppingType_Value TrainingJobEarlyStoppingType_Value_Value_MAX = TrainingJobEarlyStoppingType_Value_AUTO; +const int TrainingJobEarlyStoppingType_Value_Value_ARRAYSIZE = TrainingJobEarlyStoppingType_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* TrainingJobEarlyStoppingType_Value_descriptor(); +inline const ::std::string& TrainingJobEarlyStoppingType_Value_Name(TrainingJobEarlyStoppingType_Value value) { + return ::google::protobuf::internal::NameOfEnum( + TrainingJobEarlyStoppingType_Value_descriptor(), value); +} +inline bool TrainingJobEarlyStoppingType_Value_Parse( + const ::std::string& name, TrainingJobEarlyStoppingType_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + TrainingJobEarlyStoppingType_Value_descriptor(), name, value); +} +// =================================================================== + +class HyperparameterTuningJob final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterTuningJob) */ { + public: + HyperparameterTuningJob(); + virtual ~HyperparameterTuningJob(); + + HyperparameterTuningJob(const HyperparameterTuningJob& from); + + inline HyperparameterTuningJob& operator=(const HyperparameterTuningJob& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterTuningJob(HyperparameterTuningJob&& from) noexcept + : HyperparameterTuningJob() { + *this = ::std::move(from); + } + + inline HyperparameterTuningJob& operator=(HyperparameterTuningJob&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterTuningJob& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterTuningJob* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterTuningJob_default_instance_); + } + static constexpr int kIndexInFileMessages = + 0; + + void Swap(HyperparameterTuningJob* other); + friend void swap(HyperparameterTuningJob& a, HyperparameterTuningJob& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterTuningJob* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterTuningJob* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterTuningJob& from); + void MergeFrom(const HyperparameterTuningJob& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterTuningJob* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + bool has_training_job() const; + void clear_training_job(); + static const int kTrainingJobFieldNumber = 1; + const ::flyteidl::plugins::sagemaker::TrainingJob& training_job() const; + ::flyteidl::plugins::sagemaker::TrainingJob* release_training_job(); + ::flyteidl::plugins::sagemaker::TrainingJob* mutable_training_job(); + void set_allocated_training_job(::flyteidl::plugins::sagemaker::TrainingJob* training_job); + + // int64 max_number_of_training_jobs = 2; + void clear_max_number_of_training_jobs(); + static const int kMaxNumberOfTrainingJobsFieldNumber = 2; + ::google::protobuf::int64 max_number_of_training_jobs() const; + void set_max_number_of_training_jobs(::google::protobuf::int64 value); + + // int64 max_parallel_training_jobs = 3; + void clear_max_parallel_training_jobs(); + static const int kMaxParallelTrainingJobsFieldNumber = 3; + ::google::protobuf::int64 max_parallel_training_jobs() const; + void set_max_parallel_training_jobs(::google::protobuf::int64 value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::flyteidl::plugins::sagemaker::TrainingJob* training_job_; + ::google::protobuf::int64 max_number_of_training_jobs_; + ::google::protobuf::int64 max_parallel_training_jobs_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class HyperparameterTuningObjectiveType final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) */ { + public: + HyperparameterTuningObjectiveType(); + virtual ~HyperparameterTuningObjectiveType(); + + HyperparameterTuningObjectiveType(const HyperparameterTuningObjectiveType& from); + + inline HyperparameterTuningObjectiveType& operator=(const HyperparameterTuningObjectiveType& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterTuningObjectiveType(HyperparameterTuningObjectiveType&& from) noexcept + : HyperparameterTuningObjectiveType() { + *this = ::std::move(from); + } + + inline HyperparameterTuningObjectiveType& operator=(HyperparameterTuningObjectiveType&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterTuningObjectiveType& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterTuningObjectiveType* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterTuningObjectiveType_default_instance_); + } + static constexpr int kIndexInFileMessages = + 1; + + void Swap(HyperparameterTuningObjectiveType* other); + friend void swap(HyperparameterTuningObjectiveType& a, HyperparameterTuningObjectiveType& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterTuningObjectiveType* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterTuningObjectiveType* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterTuningObjectiveType& from); + void MergeFrom(const HyperparameterTuningObjectiveType& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterTuningObjectiveType* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef HyperparameterTuningObjectiveType_Value Value; + static const Value MINIMIZE = + HyperparameterTuningObjectiveType_Value_MINIMIZE; + static const Value MAXIMIZE = + HyperparameterTuningObjectiveType_Value_MAXIMIZE; + static inline bool Value_IsValid(int value) { + return HyperparameterTuningObjectiveType_Value_IsValid(value); + } + static const Value Value_MIN = + HyperparameterTuningObjectiveType_Value_Value_MIN; + static const Value Value_MAX = + HyperparameterTuningObjectiveType_Value_Value_MAX; + static const int Value_ARRAYSIZE = + HyperparameterTuningObjectiveType_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return HyperparameterTuningObjectiveType_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return HyperparameterTuningObjectiveType_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return HyperparameterTuningObjectiveType_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class HyperparameterTuningObjective final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) */ { + public: + HyperparameterTuningObjective(); + virtual ~HyperparameterTuningObjective(); + + HyperparameterTuningObjective(const HyperparameterTuningObjective& from); + + inline HyperparameterTuningObjective& operator=(const HyperparameterTuningObjective& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterTuningObjective(HyperparameterTuningObjective&& from) noexcept + : HyperparameterTuningObjective() { + *this = ::std::move(from); + } + + inline HyperparameterTuningObjective& operator=(HyperparameterTuningObjective&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterTuningObjective& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterTuningObjective* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterTuningObjective_default_instance_); + } + static constexpr int kIndexInFileMessages = + 2; + + void Swap(HyperparameterTuningObjective* other); + friend void swap(HyperparameterTuningObjective& a, HyperparameterTuningObjective& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterTuningObjective* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterTuningObjective* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterTuningObjective& from); + void MergeFrom(const HyperparameterTuningObjective& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterTuningObjective* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // string metric_name = 2; + void clear_metric_name(); + static const int kMetricNameFieldNumber = 2; + const ::std::string& metric_name() const; + void set_metric_name(const ::std::string& value); + #if LANG_CXX11 + void set_metric_name(::std::string&& value); + #endif + void set_metric_name(const char* value); + void set_metric_name(const char* value, size_t size); + ::std::string* mutable_metric_name(); + ::std::string* release_metric_name(); + void set_allocated_metric_name(::std::string* metric_name); + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + void clear_objective_type(); + static const int kObjectiveTypeFieldNumber = 1; + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value objective_type() const; + void set_objective_type(::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::internal::ArenaStringPtr metric_name_; + int objective_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class HyperparameterTuningStrategy final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) */ { + public: + HyperparameterTuningStrategy(); + virtual ~HyperparameterTuningStrategy(); + + HyperparameterTuningStrategy(const HyperparameterTuningStrategy& from); + + inline HyperparameterTuningStrategy& operator=(const HyperparameterTuningStrategy& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterTuningStrategy(HyperparameterTuningStrategy&& from) noexcept + : HyperparameterTuningStrategy() { + *this = ::std::move(from); + } + + inline HyperparameterTuningStrategy& operator=(HyperparameterTuningStrategy&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterTuningStrategy& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterTuningStrategy* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterTuningStrategy_default_instance_); + } + static constexpr int kIndexInFileMessages = + 3; + + void Swap(HyperparameterTuningStrategy* other); + friend void swap(HyperparameterTuningStrategy& a, HyperparameterTuningStrategy& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterTuningStrategy* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterTuningStrategy* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterTuningStrategy& from); + void MergeFrom(const HyperparameterTuningStrategy& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterTuningStrategy* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef HyperparameterTuningStrategy_Value Value; + static const Value BAYESIAN = + HyperparameterTuningStrategy_Value_BAYESIAN; + static const Value RANDOM = + HyperparameterTuningStrategy_Value_RANDOM; + static inline bool Value_IsValid(int value) { + return HyperparameterTuningStrategy_Value_IsValid(value); + } + static const Value Value_MIN = + HyperparameterTuningStrategy_Value_Value_MIN; + static const Value Value_MAX = + HyperparameterTuningStrategy_Value_Value_MAX; + static const int Value_ARRAYSIZE = + HyperparameterTuningStrategy_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return HyperparameterTuningStrategy_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return HyperparameterTuningStrategy_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return HyperparameterTuningStrategy_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class TrainingJobEarlyStoppingType final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) */ { + public: + TrainingJobEarlyStoppingType(); + virtual ~TrainingJobEarlyStoppingType(); + + TrainingJobEarlyStoppingType(const TrainingJobEarlyStoppingType& from); + + inline TrainingJobEarlyStoppingType& operator=(const TrainingJobEarlyStoppingType& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + TrainingJobEarlyStoppingType(TrainingJobEarlyStoppingType&& from) noexcept + : TrainingJobEarlyStoppingType() { + *this = ::std::move(from); + } + + inline TrainingJobEarlyStoppingType& operator=(TrainingJobEarlyStoppingType&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const TrainingJobEarlyStoppingType& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const TrainingJobEarlyStoppingType* internal_default_instance() { + return reinterpret_cast( + &_TrainingJobEarlyStoppingType_default_instance_); + } + static constexpr int kIndexInFileMessages = + 4; + + void Swap(TrainingJobEarlyStoppingType* other); + friend void swap(TrainingJobEarlyStoppingType& a, TrainingJobEarlyStoppingType& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline TrainingJobEarlyStoppingType* New() const final { + return CreateMaybeMessage(nullptr); + } + + TrainingJobEarlyStoppingType* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const TrainingJobEarlyStoppingType& from); + void MergeFrom(const TrainingJobEarlyStoppingType& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(TrainingJobEarlyStoppingType* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef TrainingJobEarlyStoppingType_Value Value; + static const Value OFF = + TrainingJobEarlyStoppingType_Value_OFF; + static const Value AUTO = + TrainingJobEarlyStoppingType_Value_AUTO; + static inline bool Value_IsValid(int value) { + return TrainingJobEarlyStoppingType_Value_IsValid(value); + } + static const Value Value_MIN = + TrainingJobEarlyStoppingType_Value_Value_MIN; + static const Value Value_MAX = + TrainingJobEarlyStoppingType_Value_Value_MAX; + static const int Value_ARRAYSIZE = + TrainingJobEarlyStoppingType_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return TrainingJobEarlyStoppingType_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return TrainingJobEarlyStoppingType_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return TrainingJobEarlyStoppingType_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class HyperparameterTuningJobConfig final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) */ { + public: + HyperparameterTuningJobConfig(); + virtual ~HyperparameterTuningJobConfig(); + + HyperparameterTuningJobConfig(const HyperparameterTuningJobConfig& from); + + inline HyperparameterTuningJobConfig& operator=(const HyperparameterTuningJobConfig& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterTuningJobConfig(HyperparameterTuningJobConfig&& from) noexcept + : HyperparameterTuningJobConfig() { + *this = ::std::move(from); + } + + inline HyperparameterTuningJobConfig& operator=(HyperparameterTuningJobConfig&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterTuningJobConfig& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterTuningJobConfig* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterTuningJobConfig_default_instance_); + } + static constexpr int kIndexInFileMessages = + 5; + + void Swap(HyperparameterTuningJobConfig* other); + friend void swap(HyperparameterTuningJobConfig& a, HyperparameterTuningJobConfig& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterTuningJobConfig* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterTuningJobConfig* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterTuningJobConfig& from); + void MergeFrom(const HyperparameterTuningJobConfig& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterTuningJobConfig* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + bool has_hyperparameter_ranges() const; + void clear_hyperparameter_ranges(); + static const int kHyperparameterRangesFieldNumber = 1; + const ::flyteidl::plugins::sagemaker::ParameterRanges& hyperparameter_ranges() const; + ::flyteidl::plugins::sagemaker::ParameterRanges* release_hyperparameter_ranges(); + ::flyteidl::plugins::sagemaker::ParameterRanges* mutable_hyperparameter_ranges(); + void set_allocated_hyperparameter_ranges(::flyteidl::plugins::sagemaker::ParameterRanges* hyperparameter_ranges); + + // .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + bool has_tuning_objective() const; + void clear_tuning_objective(); + static const int kTuningObjectiveFieldNumber = 3; + const ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective& tuning_objective() const; + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* release_tuning_objective(); + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* mutable_tuning_objective(); + void set_allocated_tuning_objective(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* tuning_objective); + + // .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + void clear_tuning_strategy(); + static const int kTuningStrategyFieldNumber = 2; + ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value tuning_strategy() const; + void set_tuning_strategy(::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value value); + + // .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + void clear_training_job_early_stopping_type(); + static const int kTrainingJobEarlyStoppingTypeFieldNumber = 4; + ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value training_job_early_stopping_type() const; + void set_training_job_early_stopping_type(::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::flyteidl::plugins::sagemaker::ParameterRanges* hyperparameter_ranges_; + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* tuning_objective_; + int tuning_strategy_; + int training_job_early_stopping_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto; +}; +// =================================================================== + + +// =================================================================== + +#ifdef __GNUC__ + #pragma GCC diagnostic push + #pragma GCC diagnostic ignored "-Wstrict-aliasing" +#endif // __GNUC__ +// HyperparameterTuningJob + +// .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; +inline bool HyperparameterTuningJob::has_training_job() const { + return this != internal_default_instance() && training_job_ != nullptr; +} +inline const ::flyteidl::plugins::sagemaker::TrainingJob& HyperparameterTuningJob::training_job() const { + const ::flyteidl::plugins::sagemaker::TrainingJob* p = training_job_; + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job) + return p != nullptr ? *p : *reinterpret_cast( + &::flyteidl::plugins::sagemaker::_TrainingJob_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::TrainingJob* HyperparameterTuningJob::release_training_job() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job) + + ::flyteidl::plugins::sagemaker::TrainingJob* temp = training_job_; + training_job_ = nullptr; + return temp; +} +inline ::flyteidl::plugins::sagemaker::TrainingJob* HyperparameterTuningJob::mutable_training_job() { + + if (training_job_ == nullptr) { + auto* p = CreateMaybeMessage<::flyteidl::plugins::sagemaker::TrainingJob>(GetArenaNoVirtual()); + training_job_ = p; + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job) + return training_job_; +} +inline void HyperparameterTuningJob::set_allocated_training_job(::flyteidl::plugins::sagemaker::TrainingJob* training_job) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + if (message_arena == nullptr) { + delete reinterpret_cast< ::google::protobuf::MessageLite*>(training_job_); + } + if (training_job) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + training_job = ::google::protobuf::internal::GetOwnedMessage( + message_arena, training_job, submessage_arena); + } + + } else { + + } + training_job_ = training_job; + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job) +} + +// int64 max_number_of_training_jobs = 2; +inline void HyperparameterTuningJob::clear_max_number_of_training_jobs() { + max_number_of_training_jobs_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 HyperparameterTuningJob::max_number_of_training_jobs() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_number_of_training_jobs) + return max_number_of_training_jobs_; +} +inline void HyperparameterTuningJob::set_max_number_of_training_jobs(::google::protobuf::int64 value) { + + max_number_of_training_jobs_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_number_of_training_jobs) +} + +// int64 max_parallel_training_jobs = 3; +inline void HyperparameterTuningJob::clear_max_parallel_training_jobs() { + max_parallel_training_jobs_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 HyperparameterTuningJob::max_parallel_training_jobs() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_parallel_training_jobs) + return max_parallel_training_jobs_; +} +inline void HyperparameterTuningJob::set_max_parallel_training_jobs(::google::protobuf::int64 value) { + + max_parallel_training_jobs_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_parallel_training_jobs) +} + +// ------------------------------------------------------------------- + +// HyperparameterTuningObjectiveType + +// ------------------------------------------------------------------- + +// HyperparameterTuningObjective + +// .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; +inline void HyperparameterTuningObjective::clear_objective_type() { + objective_type_ = 0; +} +inline ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value HyperparameterTuningObjective::objective_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.objective_type) + return static_cast< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value >(objective_type_); +} +inline void HyperparameterTuningObjective::set_objective_type(::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value value) { + + objective_type_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.objective_type) +} + +// string metric_name = 2; +inline void HyperparameterTuningObjective::clear_metric_name() { + metric_name_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline const ::std::string& HyperparameterTuningObjective::metric_name() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) + return metric_name_.GetNoArena(); +} +inline void HyperparameterTuningObjective::set_metric_name(const ::std::string& value) { + + metric_name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), value); + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) +} +#if LANG_CXX11 +inline void HyperparameterTuningObjective::set_metric_name(::std::string&& value) { + + metric_name_.SetNoArena( + &::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::move(value)); + // @@protoc_insertion_point(field_set_rvalue:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) +} +#endif +inline void HyperparameterTuningObjective::set_metric_name(const char* value) { + GOOGLE_DCHECK(value != nullptr); + + metric_name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::string(value)); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) +} +inline void HyperparameterTuningObjective::set_metric_name(const char* value, size_t size) { + + metric_name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), + ::std::string(reinterpret_cast(value), size)); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) +} +inline ::std::string* HyperparameterTuningObjective::mutable_metric_name() { + + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) + return metric_name_.MutableNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline ::std::string* HyperparameterTuningObjective::release_metric_name() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) + + return metric_name_.ReleaseNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline void HyperparameterTuningObjective::set_allocated_metric_name(::std::string* metric_name) { + if (metric_name != nullptr) { + + } else { + + } + metric_name_.SetAllocatedNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), metric_name); + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name) +} + +// ------------------------------------------------------------------- + +// HyperparameterTuningStrategy + +// ------------------------------------------------------------------- + +// TrainingJobEarlyStoppingType + +// ------------------------------------------------------------------- + +// HyperparameterTuningJobConfig + +// .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; +inline bool HyperparameterTuningJobConfig::has_hyperparameter_ranges() const { + return this != internal_default_instance() && hyperparameter_ranges_ != nullptr; +} +inline const ::flyteidl::plugins::sagemaker::ParameterRanges& HyperparameterTuningJobConfig::hyperparameter_ranges() const { + const ::flyteidl::plugins::sagemaker::ParameterRanges* p = hyperparameter_ranges_; + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges) + return p != nullptr ? *p : *reinterpret_cast( + &::flyteidl::plugins::sagemaker::_ParameterRanges_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::ParameterRanges* HyperparameterTuningJobConfig::release_hyperparameter_ranges() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges) + + ::flyteidl::plugins::sagemaker::ParameterRanges* temp = hyperparameter_ranges_; + hyperparameter_ranges_ = nullptr; + return temp; +} +inline ::flyteidl::plugins::sagemaker::ParameterRanges* HyperparameterTuningJobConfig::mutable_hyperparameter_ranges() { + + if (hyperparameter_ranges_ == nullptr) { + auto* p = CreateMaybeMessage<::flyteidl::plugins::sagemaker::ParameterRanges>(GetArenaNoVirtual()); + hyperparameter_ranges_ = p; + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges) + return hyperparameter_ranges_; +} +inline void HyperparameterTuningJobConfig::set_allocated_hyperparameter_ranges(::flyteidl::plugins::sagemaker::ParameterRanges* hyperparameter_ranges) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + if (message_arena == nullptr) { + delete reinterpret_cast< ::google::protobuf::MessageLite*>(hyperparameter_ranges_); + } + if (hyperparameter_ranges) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + hyperparameter_ranges = ::google::protobuf::internal::GetOwnedMessage( + message_arena, hyperparameter_ranges, submessage_arena); + } + + } else { + + } + hyperparameter_ranges_ = hyperparameter_ranges; + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges) +} + +// .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; +inline void HyperparameterTuningJobConfig::clear_tuning_strategy() { + tuning_strategy_ = 0; +} +inline ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value HyperparameterTuningJobConfig::tuning_strategy() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_strategy) + return static_cast< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value >(tuning_strategy_); +} +inline void HyperparameterTuningJobConfig::set_tuning_strategy(::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value value) { + + tuning_strategy_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_strategy) +} + +// .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; +inline bool HyperparameterTuningJobConfig::has_tuning_objective() const { + return this != internal_default_instance() && tuning_objective_ != nullptr; +} +inline void HyperparameterTuningJobConfig::clear_tuning_objective() { + if (GetArenaNoVirtual() == nullptr && tuning_objective_ != nullptr) { + delete tuning_objective_; + } + tuning_objective_ = nullptr; +} +inline const ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective& HyperparameterTuningJobConfig::tuning_objective() const { + const ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* p = tuning_objective_; + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective) + return p != nullptr ? *p : *reinterpret_cast( + &::flyteidl::plugins::sagemaker::_HyperparameterTuningObjective_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* HyperparameterTuningJobConfig::release_tuning_objective() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective) + + ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* temp = tuning_objective_; + tuning_objective_ = nullptr; + return temp; +} +inline ::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* HyperparameterTuningJobConfig::mutable_tuning_objective() { + + if (tuning_objective_ == nullptr) { + auto* p = CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterTuningObjective>(GetArenaNoVirtual()); + tuning_objective_ = p; + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective) + return tuning_objective_; +} +inline void HyperparameterTuningJobConfig::set_allocated_tuning_objective(::flyteidl::plugins::sagemaker::HyperparameterTuningObjective* tuning_objective) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + if (message_arena == nullptr) { + delete tuning_objective_; + } + if (tuning_objective) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + tuning_objective = ::google::protobuf::internal::GetOwnedMessage( + message_arena, tuning_objective, submessage_arena); + } + + } else { + + } + tuning_objective_ = tuning_objective; + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective) +} + +// .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; +inline void HyperparameterTuningJobConfig::clear_training_job_early_stopping_type() { + training_job_early_stopping_type_ = 0; +} +inline ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value HyperparameterTuningJobConfig::training_job_early_stopping_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.training_job_early_stopping_type) + return static_cast< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value >(training_job_early_stopping_type_); +} +inline void HyperparameterTuningJobConfig::set_training_job_early_stopping_type(::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value value) { + + training_job_early_stopping_type_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.training_job_early_stopping_type) +} + +#ifdef __GNUC__ + #pragma GCC diagnostic pop +#endif // __GNUC__ +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + + +// @@protoc_insertion_point(namespace_scope) + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + +namespace google { +namespace protobuf { + +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value>() { + return ::flyteidl::plugins::sagemaker::HyperparameterTuningObjectiveType_Value_descriptor(); +} +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value>() { + return ::flyteidl::plugins::sagemaker::HyperparameterTuningStrategy_Value_descriptor(); +} +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value>() { + return ::flyteidl::plugins::sagemaker::TrainingJobEarlyStoppingType_Value_descriptor(); +} + +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) + +#include +#endif // PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fhyperparameter_5ftuning_5fjob_2eproto diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.cc new file mode 100644 index 000000000..7c840ad69 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.cc @@ -0,0 +1,26 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +#include "flyteidl/plugins/sagemaker/parameter_ranges.pb.h" +#include "flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace flyteidl +} // namespace plugins +} // namespace sagemaker + diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.h new file mode 100644 index 000000000..c314f50a3 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.grpc.pb.h @@ -0,0 +1,49 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto +#ifndef GRPC_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto__INCLUDED +#define GRPC_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto__INCLUDED + +#include "flyteidl/plugins/sagemaker/parameter_ranges.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +namespace grpc_impl { +class Channel; +class CompletionQueue; +class ServerCompletionQueue; +} // namespace grpc_impl + +namespace grpc { +namespace experimental { +template +class MessageAllocator; +} // namespace experimental +} // namespace grpc_impl + +namespace grpc { +class ServerContext; +} // namespace grpc + +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + + +#endif // GRPC_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto__INCLUDED diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.cc new file mode 100644 index 000000000..ee2ff5646 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.cc @@ -0,0 +1,2460 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +#include "flyteidl/plugins/sagemaker/parameter_ranges.pb.h" + +#include + +#include +#include +#include +#include +#include +#include +#include +#include +// @@protoc_insertion_point(includes) +#include + +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_ContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_IntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<1> scc_info_ParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto ::google::protobuf::internal::SCCInfo<3> scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class HyperparameterScalingTypeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _HyperparameterScalingType_default_instance_; +class ContinuousParameterRangeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _ContinuousParameterRange_default_instance_; +class IntegerParameterRangeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _IntegerParameterRange_default_instance_; +class CategoricalParameterRangeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _CategoricalParameterRange_default_instance_; +class ParameterRangeOneOfDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; + const ::flyteidl::plugins::sagemaker::ContinuousParameterRange* continuous_parameter_range_; + const ::flyteidl::plugins::sagemaker::IntegerParameterRange* integer_parameter_range_; + const ::flyteidl::plugins::sagemaker::CategoricalParameterRange* categorical_parameter_range_; +} _ParameterRangeOneOf_default_instance_; +class ParameterRanges_ParameterRangeMapEntry_DoNotUseDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _ParameterRanges_ParameterRangeMapEntry_DoNotUse_default_instance_; +class ParameterRangesDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _ParameterRanges_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +static void InitDefaultsHyperparameterScalingType_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_HyperparameterScalingType_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::HyperparameterScalingType(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::HyperparameterScalingType::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_HyperparameterScalingType_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsHyperparameterScalingType_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, {}}; + +static void InitDefaultsContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_ContinuousParameterRange_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::ContinuousParameterRange(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::ContinuousParameterRange::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_ContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, {}}; + +static void InitDefaultsIntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_IntegerParameterRange_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::IntegerParameterRange(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::IntegerParameterRange::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_IntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsIntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, {}}; + +static void InitDefaultsCategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_CategoricalParameterRange_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::CategoricalParameterRange(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::CategoricalParameterRange::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsCategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, {}}; + +static void InitDefaultsParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_ParameterRangeOneOf_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::ParameterRangeOneOf(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::ParameterRangeOneOf::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<3> scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 3, InitDefaultsParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, { + &scc_info_ContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base, + &scc_info_IntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base, + &scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base,}}; + +static void InitDefaultsParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_ParameterRanges_ParameterRangeMapEntry_DoNotUse_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse(); + } + ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<1> scc_info_ParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 1, InitDefaultsParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, { + &scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base,}}; + +static void InitDefaultsParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_ParameterRanges_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::ParameterRanges(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::ParameterRanges::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<1> scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 1, InitDefaultsParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto}, { + &scc_info_ParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base,}}; + +void InitDefaults_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + ::google::protobuf::internal::InitSCC(&scc_info_HyperparameterScalingType_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_ContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_IntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_ParameterRanges_ParameterRangeMapEntry_DoNotUse_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); +} + +::google::protobuf::Metadata file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[7]; +const ::google::protobuf::EnumDescriptor* file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[1]; +constexpr ::google::protobuf::ServiceDescriptor const** file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = nullptr; + +const ::google::protobuf::uint32 TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto::offsets[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::HyperparameterScalingType, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ContinuousParameterRange, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ContinuousParameterRange, max_value_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ContinuousParameterRange, min_value_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ContinuousParameterRange, scaling_type_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::IntegerParameterRange, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::IntegerParameterRange, max_value_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::IntegerParameterRange, min_value_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::IntegerParameterRange, scaling_type_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::CategoricalParameterRange, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::CategoricalParameterRange, values_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRangeOneOf, _internal_metadata_), + ~0u, // no _extensions_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRangeOneOf, _oneof_case_[0]), + ~0u, // no _weak_field_map_ + offsetof(::flyteidl::plugins::sagemaker::ParameterRangeOneOfDefaultTypeInternal, continuous_parameter_range_), + offsetof(::flyteidl::plugins::sagemaker::ParameterRangeOneOfDefaultTypeInternal, integer_parameter_range_), + offsetof(::flyteidl::plugins::sagemaker::ParameterRangeOneOfDefaultTypeInternal, categorical_parameter_range_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRangeOneOf, parameter_range_type_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse, _has_bits_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse, key_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse, value_), + 0, + 1, + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::ParameterRanges, parameter_range_map_), +}; +static const ::google::protobuf::internal::MigrationSchema schemas[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + { 0, -1, sizeof(::flyteidl::plugins::sagemaker::HyperparameterScalingType)}, + { 5, -1, sizeof(::flyteidl::plugins::sagemaker::ContinuousParameterRange)}, + { 13, -1, sizeof(::flyteidl::plugins::sagemaker::IntegerParameterRange)}, + { 21, -1, sizeof(::flyteidl::plugins::sagemaker::CategoricalParameterRange)}, + { 27, -1, sizeof(::flyteidl::plugins::sagemaker::ParameterRangeOneOf)}, + { 36, 43, sizeof(::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse)}, + { 45, -1, sizeof(::flyteidl::plugins::sagemaker::ParameterRanges)}, +}; + +static ::google::protobuf::Message const * const file_default_instances[] = { + reinterpret_cast(&::flyteidl::plugins::sagemaker::_HyperparameterScalingType_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_ContinuousParameterRange_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_IntegerParameterRange_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_CategoricalParameterRange_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_ParameterRangeOneOf_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_ParameterRanges_ParameterRangeMapEntry_DoNotUse_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_ParameterRanges_default_instance_), +}; + +::google::protobuf::internal::AssignDescriptorsTable assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = { + {}, AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, "flyteidl/plugins/sagemaker/parameter_ranges.proto", schemas, + file_default_instances, TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto::offsets, + file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, 7, file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, +}; + +const char descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[] = + "\n1flyteidl/plugins/sagemaker/parameter_r" + "anges.proto\022\032flyteidl.plugins.sagemaker\"" + "c\n\031HyperparameterScalingType\"F\n\005Value\022\010\n" + "\004AUTO\020\000\022\n\n\006LINEAR\020\001\022\017\n\013LOGARITHMIC\020\002\022\026\n\022" + "REVERSELOGARITHMIC\020\003\"\223\001\n\030ContinuousParam" + "eterRange\022\021\n\tmax_value\030\001 \001(\001\022\021\n\tmin_valu" + "e\030\002 \001(\001\022Q\n\014scaling_type\030\003 \001(\0162;.flyteidl" + ".plugins.sagemaker.HyperparameterScaling" + "Type.Value\"\220\001\n\025IntegerParameterRange\022\021\n\t" + "max_value\030\001 \001(\003\022\021\n\tmin_value\030\002 \001(\003\022Q\n\014sc" + "aling_type\030\003 \001(\0162;.flyteidl.plugins.sage" + "maker.HyperparameterScalingType.Value\"+\n" + "\031CategoricalParameterRange\022\016\n\006values\030\001 \003" + "(\t\"\275\002\n\023ParameterRangeOneOf\022Z\n\032continuous" + "_parameter_range\030\001 \001(\01324.flyteidl.plugin" + "s.sagemaker.ContinuousParameterRangeH\000\022T" + "\n\027integer_parameter_range\030\002 \001(\01321.flytei" + "dl.plugins.sagemaker.IntegerParameterRan" + "geH\000\022\\\n\033categorical_parameter_range\030\003 \001(" + "\01325.flyteidl.plugins.sagemaker.Categoric" + "alParameterRangeH\000B\026\n\024parameter_range_ty" + "pe\"\335\001\n\017ParameterRanges\022_\n\023parameter_rang" + "e_map\030\001 \003(\0132B.flyteidl.plugins.sagemaker" + ".ParameterRanges.ParameterRangeMapEntry\032" + "i\n\026ParameterRangeMapEntry\022\013\n\003key\030\001 \001(\t\022>" + "\n\005value\030\002 \001(\0132/.flyteidl.plugins.sagemak" + "er.ParameterRangeOneOf:\0028\001B5Z3github.com" + "/lyft/flyteidl/gen/pb-go/flyteidl/plugin" + "sb\006proto3" + ; +::google::protobuf::internal::DescriptorTable descriptor_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = { + false, InitDefaults_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, + descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, + "flyteidl/plugins/sagemaker/parameter_ranges.proto", &assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, 1129, +}; + +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto() { + static constexpr ::google::protobuf::internal::InitFunc deps[1] = + { + }; + ::google::protobuf::internal::AddDescriptors(&descriptor_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto, deps, 0); +} + +// Force running AddDescriptors() at dynamic initialization time. +static bool dynamic_init_dummy_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto = []() { AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto(); return true; }(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +const ::google::protobuf::EnumDescriptor* HyperparameterScalingType_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[0]; +} +bool HyperparameterScalingType_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + case 2: + case 3: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const HyperparameterScalingType_Value HyperparameterScalingType::AUTO; +const HyperparameterScalingType_Value HyperparameterScalingType::LINEAR; +const HyperparameterScalingType_Value HyperparameterScalingType::LOGARITHMIC; +const HyperparameterScalingType_Value HyperparameterScalingType::REVERSELOGARITHMIC; +const HyperparameterScalingType_Value HyperparameterScalingType::Value_MIN; +const HyperparameterScalingType_Value HyperparameterScalingType::Value_MAX; +const int HyperparameterScalingType::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +// =================================================================== + +void HyperparameterScalingType::InitAsDefaultInstance() { +} +class HyperparameterScalingType::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +HyperparameterScalingType::HyperparameterScalingType() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.HyperparameterScalingType) +} +HyperparameterScalingType::HyperparameterScalingType(const HyperparameterScalingType& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.HyperparameterScalingType) +} + +void HyperparameterScalingType::SharedCtor() { +} + +HyperparameterScalingType::~HyperparameterScalingType() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.HyperparameterScalingType) + SharedDtor(); +} + +void HyperparameterScalingType::SharedDtor() { +} + +void HyperparameterScalingType::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const HyperparameterScalingType& HyperparameterScalingType::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_HyperparameterScalingType_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void HyperparameterScalingType::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* HyperparameterScalingType::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool HyperparameterScalingType::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.HyperparameterScalingType) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.HyperparameterScalingType) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void HyperparameterScalingType::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.HyperparameterScalingType) +} + +::google::protobuf::uint8* HyperparameterScalingType::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.HyperparameterScalingType) + return target; +} + +size_t HyperparameterScalingType::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void HyperparameterScalingType::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + GOOGLE_DCHECK_NE(&from, this); + const HyperparameterScalingType* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.HyperparameterScalingType) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.HyperparameterScalingType) + MergeFrom(*source); + } +} + +void HyperparameterScalingType::MergeFrom(const HyperparameterScalingType& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void HyperparameterScalingType::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void HyperparameterScalingType::CopyFrom(const HyperparameterScalingType& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.HyperparameterScalingType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool HyperparameterScalingType::IsInitialized() const { + return true; +} + +void HyperparameterScalingType::Swap(HyperparameterScalingType* other) { + if (other == this) return; + InternalSwap(other); +} +void HyperparameterScalingType::InternalSwap(HyperparameterScalingType* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata HyperparameterScalingType::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void ContinuousParameterRange::InitAsDefaultInstance() { +} +class ContinuousParameterRange::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int ContinuousParameterRange::kMaxValueFieldNumber; +const int ContinuousParameterRange::kMinValueFieldNumber; +const int ContinuousParameterRange::kScalingTypeFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +ContinuousParameterRange::ContinuousParameterRange() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.ContinuousParameterRange) +} +ContinuousParameterRange::ContinuousParameterRange(const ContinuousParameterRange& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::memcpy(&max_value_, &from.max_value_, + static_cast(reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.ContinuousParameterRange) +} + +void ContinuousParameterRange::SharedCtor() { + ::memset(&max_value_, 0, static_cast( + reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); +} + +ContinuousParameterRange::~ContinuousParameterRange() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.ContinuousParameterRange) + SharedDtor(); +} + +void ContinuousParameterRange::SharedDtor() { +} + +void ContinuousParameterRange::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const ContinuousParameterRange& ContinuousParameterRange::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_ContinuousParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void ContinuousParameterRange::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + ::memset(&max_value_, 0, static_cast( + reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* ContinuousParameterRange::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // double max_value = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 9) goto handle_unusual; + msg->set_max_value(::google::protobuf::io::UnalignedLoad(ptr)); + ptr += sizeof(double); + break; + } + // double min_value = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 17) goto handle_unusual; + msg->set_min_value(::google::protobuf::io::UnalignedLoad(ptr)); + ptr += sizeof(double); + break; + } + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 24) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_scaling_type(static_cast<::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool ContinuousParameterRange::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // double max_value = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (9 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + double, ::google::protobuf::internal::WireFormatLite::TYPE_DOUBLE>( + input, &max_value_))); + } else { + goto handle_unusual; + } + break; + } + + // double min_value = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (17 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + double, ::google::protobuf::internal::WireFormatLite::TYPE_DOUBLE>( + input, &min_value_))); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (24 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_scaling_type(static_cast< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.ContinuousParameterRange) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.ContinuousParameterRange) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void ContinuousParameterRange::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // double max_value = 1; + if (this->max_value() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteDouble(1, this->max_value(), output); + } + + // double min_value = 2; + if (this->min_value() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteDouble(2, this->min_value(), output); + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 3, this->scaling_type(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.ContinuousParameterRange) +} + +::google::protobuf::uint8* ContinuousParameterRange::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // double max_value = 1; + if (this->max_value() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteDoubleToArray(1, this->max_value(), target); + } + + // double min_value = 2; + if (this->min_value() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteDoubleToArray(2, this->min_value(), target); + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 3, this->scaling_type(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.ContinuousParameterRange) + return target; +} + +size_t ContinuousParameterRange::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // double max_value = 1; + if (this->max_value() != 0) { + total_size += 1 + 8; + } + + // double min_value = 2; + if (this->min_value() != 0) { + total_size += 1 + 8; + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->scaling_type()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void ContinuousParameterRange::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + GOOGLE_DCHECK_NE(&from, this); + const ContinuousParameterRange* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.ContinuousParameterRange) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.ContinuousParameterRange) + MergeFrom(*source); + } +} + +void ContinuousParameterRange::MergeFrom(const ContinuousParameterRange& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.max_value() != 0) { + set_max_value(from.max_value()); + } + if (from.min_value() != 0) { + set_min_value(from.min_value()); + } + if (from.scaling_type() != 0) { + set_scaling_type(from.scaling_type()); + } +} + +void ContinuousParameterRange::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void ContinuousParameterRange::CopyFrom(const ContinuousParameterRange& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.ContinuousParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool ContinuousParameterRange::IsInitialized() const { + return true; +} + +void ContinuousParameterRange::Swap(ContinuousParameterRange* other) { + if (other == this) return; + InternalSwap(other); +} +void ContinuousParameterRange::InternalSwap(ContinuousParameterRange* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(max_value_, other->max_value_); + swap(min_value_, other->min_value_); + swap(scaling_type_, other->scaling_type_); +} + +::google::protobuf::Metadata ContinuousParameterRange::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void IntegerParameterRange::InitAsDefaultInstance() { +} +class IntegerParameterRange::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int IntegerParameterRange::kMaxValueFieldNumber; +const int IntegerParameterRange::kMinValueFieldNumber; +const int IntegerParameterRange::kScalingTypeFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +IntegerParameterRange::IntegerParameterRange() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.IntegerParameterRange) +} +IntegerParameterRange::IntegerParameterRange(const IntegerParameterRange& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::memcpy(&max_value_, &from.max_value_, + static_cast(reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.IntegerParameterRange) +} + +void IntegerParameterRange::SharedCtor() { + ::memset(&max_value_, 0, static_cast( + reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); +} + +IntegerParameterRange::~IntegerParameterRange() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.IntegerParameterRange) + SharedDtor(); +} + +void IntegerParameterRange::SharedDtor() { +} + +void IntegerParameterRange::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const IntegerParameterRange& IntegerParameterRange::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_IntegerParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void IntegerParameterRange::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + ::memset(&max_value_, 0, static_cast( + reinterpret_cast(&scaling_type_) - + reinterpret_cast(&max_value_)) + sizeof(scaling_type_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* IntegerParameterRange::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // int64 max_value = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 8) goto handle_unusual; + msg->set_max_value(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // int64 min_value = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 16) goto handle_unusual; + msg->set_min_value(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 24) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_scaling_type(static_cast<::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool IntegerParameterRange::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // int64 max_value = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (8 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &max_value_))); + } else { + goto handle_unusual; + } + break; + } + + // int64 min_value = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (16 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &min_value_))); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (24 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_scaling_type(static_cast< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.IntegerParameterRange) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.IntegerParameterRange) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void IntegerParameterRange::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // int64 max_value = 1; + if (this->max_value() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(1, this->max_value(), output); + } + + // int64 min_value = 2; + if (this->min_value() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(2, this->min_value(), output); + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 3, this->scaling_type(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.IntegerParameterRange) +} + +::google::protobuf::uint8* IntegerParameterRange::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // int64 max_value = 1; + if (this->max_value() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(1, this->max_value(), target); + } + + // int64 min_value = 2; + if (this->min_value() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(2, this->min_value(), target); + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 3, this->scaling_type(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.IntegerParameterRange) + return target; +} + +size_t IntegerParameterRange::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // int64 max_value = 1; + if (this->max_value() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->max_value()); + } + + // int64 min_value = 2; + if (this->min_value() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->min_value()); + } + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + if (this->scaling_type() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->scaling_type()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void IntegerParameterRange::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + GOOGLE_DCHECK_NE(&from, this); + const IntegerParameterRange* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.IntegerParameterRange) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.IntegerParameterRange) + MergeFrom(*source); + } +} + +void IntegerParameterRange::MergeFrom(const IntegerParameterRange& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.max_value() != 0) { + set_max_value(from.max_value()); + } + if (from.min_value() != 0) { + set_min_value(from.min_value()); + } + if (from.scaling_type() != 0) { + set_scaling_type(from.scaling_type()); + } +} + +void IntegerParameterRange::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void IntegerParameterRange::CopyFrom(const IntegerParameterRange& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.IntegerParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool IntegerParameterRange::IsInitialized() const { + return true; +} + +void IntegerParameterRange::Swap(IntegerParameterRange* other) { + if (other == this) return; + InternalSwap(other); +} +void IntegerParameterRange::InternalSwap(IntegerParameterRange* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(max_value_, other->max_value_); + swap(min_value_, other->min_value_); + swap(scaling_type_, other->scaling_type_); +} + +::google::protobuf::Metadata IntegerParameterRange::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void CategoricalParameterRange::InitAsDefaultInstance() { +} +class CategoricalParameterRange::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int CategoricalParameterRange::kValuesFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +CategoricalParameterRange::CategoricalParameterRange() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.CategoricalParameterRange) +} +CategoricalParameterRange::CategoricalParameterRange(const CategoricalParameterRange& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr), + values_(from.values_) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.CategoricalParameterRange) +} + +void CategoricalParameterRange::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); +} + +CategoricalParameterRange::~CategoricalParameterRange() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.CategoricalParameterRange) + SharedDtor(); +} + +void CategoricalParameterRange::SharedDtor() { +} + +void CategoricalParameterRange::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const CategoricalParameterRange& CategoricalParameterRange::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_CategoricalParameterRange_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void CategoricalParameterRange::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + values_.Clear(); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* CategoricalParameterRange::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // repeated string values = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + do { + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.CategoricalParameterRange.values"); + object = msg->add_values(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + if (ptr >= end) break; + } while ((::google::protobuf::io::UnalignedLoad<::google::protobuf::uint64>(ptr) & 255) == 10 && (ptr += 1)); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +string_till_end: + static_cast<::std::string*>(object)->clear(); + static_cast<::std::string*>(object)->reserve(size); + goto len_delim_till_end; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool CategoricalParameterRange::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // repeated string values = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->add_values())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->values(this->values_size() - 1).data(), + static_cast(this->values(this->values_size() - 1).length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.CategoricalParameterRange.values")); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.CategoricalParameterRange) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.CategoricalParameterRange) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void CategoricalParameterRange::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // repeated string values = 1; + for (int i = 0, n = this->values_size(); i < n; i++) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->values(i).data(), static_cast(this->values(i).length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.CategoricalParameterRange.values"); + ::google::protobuf::internal::WireFormatLite::WriteString( + 1, this->values(i), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.CategoricalParameterRange) +} + +::google::protobuf::uint8* CategoricalParameterRange::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // repeated string values = 1; + for (int i = 0, n = this->values_size(); i < n; i++) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->values(i).data(), static_cast(this->values(i).length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.CategoricalParameterRange.values"); + target = ::google::protobuf::internal::WireFormatLite:: + WriteStringToArray(1, this->values(i), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.CategoricalParameterRange) + return target; +} + +size_t CategoricalParameterRange::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // repeated string values = 1; + total_size += 1 * + ::google::protobuf::internal::FromIntSize(this->values_size()); + for (int i = 0, n = this->values_size(); i < n; i++) { + total_size += ::google::protobuf::internal::WireFormatLite::StringSize( + this->values(i)); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void CategoricalParameterRange::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + GOOGLE_DCHECK_NE(&from, this); + const CategoricalParameterRange* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.CategoricalParameterRange) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.CategoricalParameterRange) + MergeFrom(*source); + } +} + +void CategoricalParameterRange::MergeFrom(const CategoricalParameterRange& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + values_.MergeFrom(from.values_); +} + +void CategoricalParameterRange::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void CategoricalParameterRange::CopyFrom(const CategoricalParameterRange& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.CategoricalParameterRange) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool CategoricalParameterRange::IsInitialized() const { + return true; +} + +void CategoricalParameterRange::Swap(CategoricalParameterRange* other) { + if (other == this) return; + InternalSwap(other); +} +void CategoricalParameterRange::InternalSwap(CategoricalParameterRange* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + values_.InternalSwap(CastToBase(&other->values_)); +} + +::google::protobuf::Metadata CategoricalParameterRange::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void ParameterRangeOneOf::InitAsDefaultInstance() { + ::flyteidl::plugins::sagemaker::_ParameterRangeOneOf_default_instance_.continuous_parameter_range_ = const_cast< ::flyteidl::plugins::sagemaker::ContinuousParameterRange*>( + ::flyteidl::plugins::sagemaker::ContinuousParameterRange::internal_default_instance()); + ::flyteidl::plugins::sagemaker::_ParameterRangeOneOf_default_instance_.integer_parameter_range_ = const_cast< ::flyteidl::plugins::sagemaker::IntegerParameterRange*>( + ::flyteidl::plugins::sagemaker::IntegerParameterRange::internal_default_instance()); + ::flyteidl::plugins::sagemaker::_ParameterRangeOneOf_default_instance_.categorical_parameter_range_ = const_cast< ::flyteidl::plugins::sagemaker::CategoricalParameterRange*>( + ::flyteidl::plugins::sagemaker::CategoricalParameterRange::internal_default_instance()); +} +class ParameterRangeOneOf::HasBitSetters { + public: + static const ::flyteidl::plugins::sagemaker::ContinuousParameterRange& continuous_parameter_range(const ParameterRangeOneOf* msg); + static const ::flyteidl::plugins::sagemaker::IntegerParameterRange& integer_parameter_range(const ParameterRangeOneOf* msg); + static const ::flyteidl::plugins::sagemaker::CategoricalParameterRange& categorical_parameter_range(const ParameterRangeOneOf* msg); +}; + +const ::flyteidl::plugins::sagemaker::ContinuousParameterRange& +ParameterRangeOneOf::HasBitSetters::continuous_parameter_range(const ParameterRangeOneOf* msg) { + return *msg->parameter_range_type_.continuous_parameter_range_; +} +const ::flyteidl::plugins::sagemaker::IntegerParameterRange& +ParameterRangeOneOf::HasBitSetters::integer_parameter_range(const ParameterRangeOneOf* msg) { + return *msg->parameter_range_type_.integer_parameter_range_; +} +const ::flyteidl::plugins::sagemaker::CategoricalParameterRange& +ParameterRangeOneOf::HasBitSetters::categorical_parameter_range(const ParameterRangeOneOf* msg) { + return *msg->parameter_range_type_.categorical_parameter_range_; +} +void ParameterRangeOneOf::set_allocated_continuous_parameter_range(::flyteidl::plugins::sagemaker::ContinuousParameterRange* continuous_parameter_range) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + clear_parameter_range_type(); + if (continuous_parameter_range) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + continuous_parameter_range = ::google::protobuf::internal::GetOwnedMessage( + message_arena, continuous_parameter_range, submessage_arena); + } + set_has_continuous_parameter_range(); + parameter_range_type_.continuous_parameter_range_ = continuous_parameter_range; + } + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range) +} +void ParameterRangeOneOf::set_allocated_integer_parameter_range(::flyteidl::plugins::sagemaker::IntegerParameterRange* integer_parameter_range) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + clear_parameter_range_type(); + if (integer_parameter_range) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + integer_parameter_range = ::google::protobuf::internal::GetOwnedMessage( + message_arena, integer_parameter_range, submessage_arena); + } + set_has_integer_parameter_range(); + parameter_range_type_.integer_parameter_range_ = integer_parameter_range; + } + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range) +} +void ParameterRangeOneOf::set_allocated_categorical_parameter_range(::flyteidl::plugins::sagemaker::CategoricalParameterRange* categorical_parameter_range) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + clear_parameter_range_type(); + if (categorical_parameter_range) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + categorical_parameter_range = ::google::protobuf::internal::GetOwnedMessage( + message_arena, categorical_parameter_range, submessage_arena); + } + set_has_categorical_parameter_range(); + parameter_range_type_.categorical_parameter_range_ = categorical_parameter_range; + } + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range) +} +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int ParameterRangeOneOf::kContinuousParameterRangeFieldNumber; +const int ParameterRangeOneOf::kIntegerParameterRangeFieldNumber; +const int ParameterRangeOneOf::kCategoricalParameterRangeFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +ParameterRangeOneOf::ParameterRangeOneOf() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.ParameterRangeOneOf) +} +ParameterRangeOneOf::ParameterRangeOneOf(const ParameterRangeOneOf& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + clear_has_parameter_range_type(); + switch (from.parameter_range_type_case()) { + case kContinuousParameterRange: { + mutable_continuous_parameter_range()->::flyteidl::plugins::sagemaker::ContinuousParameterRange::MergeFrom(from.continuous_parameter_range()); + break; + } + case kIntegerParameterRange: { + mutable_integer_parameter_range()->::flyteidl::plugins::sagemaker::IntegerParameterRange::MergeFrom(from.integer_parameter_range()); + break; + } + case kCategoricalParameterRange: { + mutable_categorical_parameter_range()->::flyteidl::plugins::sagemaker::CategoricalParameterRange::MergeFrom(from.categorical_parameter_range()); + break; + } + case PARAMETER_RANGE_TYPE_NOT_SET: { + break; + } + } + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.ParameterRangeOneOf) +} + +void ParameterRangeOneOf::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + clear_has_parameter_range_type(); +} + +ParameterRangeOneOf::~ParameterRangeOneOf() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + SharedDtor(); +} + +void ParameterRangeOneOf::SharedDtor() { + if (has_parameter_range_type()) { + clear_parameter_range_type(); + } +} + +void ParameterRangeOneOf::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const ParameterRangeOneOf& ParameterRangeOneOf::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_ParameterRangeOneOf_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void ParameterRangeOneOf::clear_parameter_range_type() { +// @@protoc_insertion_point(one_of_clear_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + switch (parameter_range_type_case()) { + case kContinuousParameterRange: { + delete parameter_range_type_.continuous_parameter_range_; + break; + } + case kIntegerParameterRange: { + delete parameter_range_type_.integer_parameter_range_; + break; + } + case kCategoricalParameterRange: { + delete parameter_range_type_.categorical_parameter_range_; + break; + } + case PARAMETER_RANGE_TYPE_NOT_SET: { + break; + } + } + _oneof_case_[0] = PARAMETER_RANGE_TYPE_NOT_SET; +} + + +void ParameterRangeOneOf::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + clear_parameter_range_type(); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* ParameterRangeOneOf::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::ContinuousParameterRange::_InternalParse; + object = msg->mutable_continuous_parameter_range(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 18) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::IntegerParameterRange::_InternalParse; + object = msg->mutable_integer_parameter_range(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 26) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::CategoricalParameterRange::_InternalParse; + object = msg->mutable_categorical_parameter_range(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool ParameterRangeOneOf::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_continuous_parameter_range())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (18 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_integer_parameter_range())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (26 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_categorical_parameter_range())); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void ParameterRangeOneOf::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + if (has_continuous_parameter_range()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 1, HasBitSetters::continuous_parameter_range(this), output); + } + + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + if (has_integer_parameter_range()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 2, HasBitSetters::integer_parameter_range(this), output); + } + + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + if (has_categorical_parameter_range()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 3, HasBitSetters::categorical_parameter_range(this), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.ParameterRangeOneOf) +} + +::google::protobuf::uint8* ParameterRangeOneOf::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + if (has_continuous_parameter_range()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 1, HasBitSetters::continuous_parameter_range(this), target); + } + + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + if (has_integer_parameter_range()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 2, HasBitSetters::integer_parameter_range(this), target); + } + + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + if (has_categorical_parameter_range()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 3, HasBitSetters::categorical_parameter_range(this), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + return target; +} + +size_t ParameterRangeOneOf::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + switch (parameter_range_type_case()) { + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + case kContinuousParameterRange: { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *parameter_range_type_.continuous_parameter_range_); + break; + } + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + case kIntegerParameterRange: { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *parameter_range_type_.integer_parameter_range_); + break; + } + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + case kCategoricalParameterRange: { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *parameter_range_type_.categorical_parameter_range_); + break; + } + case PARAMETER_RANGE_TYPE_NOT_SET: { + break; + } + } + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void ParameterRangeOneOf::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + GOOGLE_DCHECK_NE(&from, this); + const ParameterRangeOneOf* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + MergeFrom(*source); + } +} + +void ParameterRangeOneOf::MergeFrom(const ParameterRangeOneOf& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + switch (from.parameter_range_type_case()) { + case kContinuousParameterRange: { + mutable_continuous_parameter_range()->::flyteidl::plugins::sagemaker::ContinuousParameterRange::MergeFrom(from.continuous_parameter_range()); + break; + } + case kIntegerParameterRange: { + mutable_integer_parameter_range()->::flyteidl::plugins::sagemaker::IntegerParameterRange::MergeFrom(from.integer_parameter_range()); + break; + } + case kCategoricalParameterRange: { + mutable_categorical_parameter_range()->::flyteidl::plugins::sagemaker::CategoricalParameterRange::MergeFrom(from.categorical_parameter_range()); + break; + } + case PARAMETER_RANGE_TYPE_NOT_SET: { + break; + } + } +} + +void ParameterRangeOneOf::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void ParameterRangeOneOf::CopyFrom(const ParameterRangeOneOf& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool ParameterRangeOneOf::IsInitialized() const { + return true; +} + +void ParameterRangeOneOf::Swap(ParameterRangeOneOf* other) { + if (other == this) return; + InternalSwap(other); +} +void ParameterRangeOneOf::InternalSwap(ParameterRangeOneOf* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(parameter_range_type_, other->parameter_range_type_); + swap(_oneof_case_[0], other->_oneof_case_[0]); +} + +::google::protobuf::Metadata ParameterRangeOneOf::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +ParameterRanges_ParameterRangeMapEntry_DoNotUse::ParameterRanges_ParameterRangeMapEntry_DoNotUse() {} +ParameterRanges_ParameterRangeMapEntry_DoNotUse::ParameterRanges_ParameterRangeMapEntry_DoNotUse(::google::protobuf::Arena* arena) + : SuperType(arena) {} +void ParameterRanges_ParameterRangeMapEntry_DoNotUse::MergeFrom(const ParameterRanges_ParameterRangeMapEntry_DoNotUse& other) { + MergeFromInternal(other); +} +::google::protobuf::Metadata ParameterRanges_ParameterRangeMapEntry_DoNotUse::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[5]; +} +void ParameterRanges_ParameterRangeMapEntry_DoNotUse::MergeFrom( + const ::google::protobuf::Message& other) { + ::google::protobuf::Message::MergeFrom(other); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool ParameterRanges_ParameterRangeMapEntry_DoNotUse::_ParseMap(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx) { + using MF = ::google::protobuf::internal::MapField< + ParameterRanges_ParameterRangeMapEntry_DoNotUse, EntryKeyType, EntryValueType, + kEntryKeyFieldType, kEntryValueFieldType, + kEntryDefaultEnumValue>; + auto mf = static_cast(object); + Parser> parser(mf); +#define DO_(x) if (!(x)) return false + DO_(parser.ParseMap(begin, end)); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + parser.key().data(), static_cast(parser.key().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.key")); +#undef DO_ + return true; +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + + +// =================================================================== + +void ParameterRanges::InitAsDefaultInstance() { +} +class ParameterRanges::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int ParameterRanges::kParameterRangeMapFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +ParameterRanges::ParameterRanges() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.ParameterRanges) +} +ParameterRanges::ParameterRanges(const ParameterRanges& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + parameter_range_map_.MergeFrom(from.parameter_range_map_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.ParameterRanges) +} + +void ParameterRanges::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); +} + +ParameterRanges::~ParameterRanges() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.ParameterRanges) + SharedDtor(); +} + +void ParameterRanges::SharedDtor() { +} + +void ParameterRanges::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const ParameterRanges& ParameterRanges::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_ParameterRanges_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto.base); + return *internal_default_instance(); +} + + +void ParameterRanges::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.ParameterRanges) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + parameter_range_map_.Clear(); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* ParameterRanges::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // map parameter_range_map = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + do { + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::google::protobuf::internal::SlowMapEntryParser; + auto parse_map = ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse::_ParseMap; + ctx->extra_parse_data().payload.clear(); + ctx->extra_parse_data().parse_map = parse_map; + object = &msg->parameter_range_map_; + if (size > end - ptr) goto len_delim_till_end; + auto newend = ptr + size; + GOOGLE_PROTOBUF_PARSER_ASSERT(parse_map(ptr, newend, object, ctx)); + ptr = newend; + if (ptr >= end) break; + } while ((::google::protobuf::io::UnalignedLoad<::google::protobuf::uint64>(ptr) & 255) == 10 && (ptr += 1)); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool ParameterRanges::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.ParameterRanges) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // map parameter_range_map = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + ParameterRanges_ParameterRangeMapEntry_DoNotUse::Parser< ::google::protobuf::internal::MapField< + ParameterRanges_ParameterRangeMapEntry_DoNotUse, + ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf, + ::google::protobuf::internal::WireFormatLite::TYPE_STRING, + ::google::protobuf::internal::WireFormatLite::TYPE_MESSAGE, + 0 >, + ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf > > parser(¶meter_range_map_); + DO_(::google::protobuf::internal::WireFormatLite::ReadMessageNoVirtual( + input, &parser)); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + parser.key().data(), static_cast(parser.key().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.key")); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.ParameterRanges) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.ParameterRanges) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void ParameterRanges::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.ParameterRanges) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // map parameter_range_map = 1; + if (!this->parameter_range_map().empty()) { + typedef ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_pointer + ConstPtr; + typedef ConstPtr SortItem; + typedef ::google::protobuf::internal::CompareByDerefFirst Less; + struct Utf8Check { + static void Check(ConstPtr p) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + p->first.data(), static_cast(p->first.length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.key"); + } + }; + + if (output->IsSerializationDeterministic() && + this->parameter_range_map().size() > 1) { + ::std::unique_ptr items( + new SortItem[this->parameter_range_map().size()]); + typedef ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::size_type size_type; + size_type n = 0; + for (::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_iterator + it = this->parameter_range_map().begin(); + it != this->parameter_range_map().end(); ++it, ++n) { + items[static_cast(n)] = SortItem(&*it); + } + ::std::sort(&items[0], &items[static_cast(n)], Less()); + ::std::unique_ptr entry; + for (size_type i = 0; i < n; i++) { + entry.reset(parameter_range_map_.NewEntryWrapper(items[static_cast(i)]->first, items[static_cast(i)]->second)); + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray(1, *entry, output); + Utf8Check::Check(&(*items[static_cast(i)])); + } + } else { + ::std::unique_ptr entry; + for (::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_iterator + it = this->parameter_range_map().begin(); + it != this->parameter_range_map().end(); ++it) { + entry.reset(parameter_range_map_.NewEntryWrapper(it->first, it->second)); + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray(1, *entry, output); + Utf8Check::Check(&(*it)); + } + } + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.ParameterRanges) +} + +::google::protobuf::uint8* ParameterRanges::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.ParameterRanges) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // map parameter_range_map = 1; + if (!this->parameter_range_map().empty()) { + typedef ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_pointer + ConstPtr; + typedef ConstPtr SortItem; + typedef ::google::protobuf::internal::CompareByDerefFirst Less; + struct Utf8Check { + static void Check(ConstPtr p) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + p->first.data(), static_cast(p->first.length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.key"); + } + }; + + if (false && + this->parameter_range_map().size() > 1) { + ::std::unique_ptr items( + new SortItem[this->parameter_range_map().size()]); + typedef ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::size_type size_type; + size_type n = 0; + for (::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_iterator + it = this->parameter_range_map().begin(); + it != this->parameter_range_map().end(); ++it, ++n) { + items[static_cast(n)] = SortItem(&*it); + } + ::std::sort(&items[0], &items[static_cast(n)], Less()); + ::std::unique_ptr entry; + for (size_type i = 0; i < n; i++) { + entry.reset(parameter_range_map_.NewEntryWrapper(items[static_cast(i)]->first, items[static_cast(i)]->second)); + target = ::google::protobuf::internal::WireFormatLite::InternalWriteMessageNoVirtualToArray(1, *entry, target); + Utf8Check::Check(&(*items[static_cast(i)])); + } + } else { + ::std::unique_ptr entry; + for (::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_iterator + it = this->parameter_range_map().begin(); + it != this->parameter_range_map().end(); ++it) { + entry.reset(parameter_range_map_.NewEntryWrapper(it->first, it->second)); + target = ::google::protobuf::internal::WireFormatLite::InternalWriteMessageNoVirtualToArray(1, *entry, target); + Utf8Check::Check(&(*it)); + } + } + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.ParameterRanges) + return target; +} + +size_t ParameterRanges::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.ParameterRanges) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // map parameter_range_map = 1; + total_size += 1 * + ::google::protobuf::internal::FromIntSize(this->parameter_range_map_size()); + { + ::std::unique_ptr entry; + for (::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >::const_iterator + it = this->parameter_range_map().begin(); + it != this->parameter_range_map().end(); ++it) { + entry.reset(parameter_range_map_.NewEntryWrapper(it->first, it->second)); + total_size += ::google::protobuf::internal::WireFormatLite:: + MessageSizeNoVirtual(*entry); + } + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void ParameterRanges::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.ParameterRanges) + GOOGLE_DCHECK_NE(&from, this); + const ParameterRanges* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.ParameterRanges) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.ParameterRanges) + MergeFrom(*source); + } +} + +void ParameterRanges::MergeFrom(const ParameterRanges& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.ParameterRanges) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + parameter_range_map_.MergeFrom(from.parameter_range_map_); +} + +void ParameterRanges::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.ParameterRanges) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void ParameterRanges::CopyFrom(const ParameterRanges& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.ParameterRanges) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool ParameterRanges::IsInitialized() const { + return true; +} + +void ParameterRanges::Swap(ParameterRanges* other) { + if (other == this) return; + InternalSwap(other); +} +void ParameterRanges::InternalSwap(ParameterRanges* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + parameter_range_map_.Swap(&other->parameter_range_map_); +} + +::google::protobuf::Metadata ParameterRanges::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto[kIndexInFileMessages]; +} + + +// @@protoc_insertion_point(namespace_scope) +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::HyperparameterScalingType* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::HyperparameterScalingType >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::HyperparameterScalingType >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::ContinuousParameterRange* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::ContinuousParameterRange >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::ContinuousParameterRange >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::IntegerParameterRange* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::IntegerParameterRange >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::IntegerParameterRange >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::CategoricalParameterRange* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::CategoricalParameterRange >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::CategoricalParameterRange >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::ParameterRangeOneOf* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::ParameterRanges* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::ParameterRanges >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::ParameterRanges >(arena); +} +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) +#include diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.h new file mode 100644 index 000000000..13201549d --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/parameter_ranges.pb.h @@ -0,0 +1,1308 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +#ifndef PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto +#define PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto + +#include +#include + +#include +#if PROTOBUF_VERSION < 3007000 +#error This file was generated by a newer version of protoc which is +#error incompatible with your Protocol Buffer headers. Please update +#error your headers. +#endif +#if 3007000 < PROTOBUF_MIN_PROTOC_VERSION +#error This file was generated by an older version of protoc which is +#error incompatible with your Protocol Buffer headers. Please +#error regenerate this file with a newer version of protoc. +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include // IWYU pragma: export +#include // IWYU pragma: export +#include // IWYU pragma: export +#include +#include +#include +#include +// @@protoc_insertion_point(includes) +#include +#define PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto + +// Internal implementation detail -- do not use these members. +struct TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto { + static const ::google::protobuf::internal::ParseTableField entries[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::AuxillaryParseTableField aux[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::ParseTable schema[7] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::FieldMetadata field_metadata[]; + static const ::google::protobuf::internal::SerializationTable serialization_table[]; + static const ::google::protobuf::uint32 offsets[]; +}; +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class CategoricalParameterRange; +class CategoricalParameterRangeDefaultTypeInternal; +extern CategoricalParameterRangeDefaultTypeInternal _CategoricalParameterRange_default_instance_; +class ContinuousParameterRange; +class ContinuousParameterRangeDefaultTypeInternal; +extern ContinuousParameterRangeDefaultTypeInternal _ContinuousParameterRange_default_instance_; +class HyperparameterScalingType; +class HyperparameterScalingTypeDefaultTypeInternal; +extern HyperparameterScalingTypeDefaultTypeInternal _HyperparameterScalingType_default_instance_; +class IntegerParameterRange; +class IntegerParameterRangeDefaultTypeInternal; +extern IntegerParameterRangeDefaultTypeInternal _IntegerParameterRange_default_instance_; +class ParameterRangeOneOf; +class ParameterRangeOneOfDefaultTypeInternal; +extern ParameterRangeOneOfDefaultTypeInternal _ParameterRangeOneOf_default_instance_; +class ParameterRanges; +class ParameterRangesDefaultTypeInternal; +extern ParameterRangesDefaultTypeInternal _ParameterRanges_default_instance_; +class ParameterRanges_ParameterRangeMapEntry_DoNotUse; +class ParameterRanges_ParameterRangeMapEntry_DoNotUseDefaultTypeInternal; +extern ParameterRanges_ParameterRangeMapEntry_DoNotUseDefaultTypeInternal _ParameterRanges_ParameterRangeMapEntry_DoNotUse_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> ::flyteidl::plugins::sagemaker::CategoricalParameterRange* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::CategoricalParameterRange>(Arena*); +template<> ::flyteidl::plugins::sagemaker::ContinuousParameterRange* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::ContinuousParameterRange>(Arena*); +template<> ::flyteidl::plugins::sagemaker::HyperparameterScalingType* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::HyperparameterScalingType>(Arena*); +template<> ::flyteidl::plugins::sagemaker::IntegerParameterRange* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::IntegerParameterRange>(Arena*); +template<> ::flyteidl::plugins::sagemaker::ParameterRangeOneOf* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::ParameterRangeOneOf>(Arena*); +template<> ::flyteidl::plugins::sagemaker::ParameterRanges* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::ParameterRanges>(Arena*); +template<> ::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::ParameterRanges_ParameterRangeMapEntry_DoNotUse>(Arena*); +} // namespace protobuf +} // namespace google +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +enum HyperparameterScalingType_Value { + HyperparameterScalingType_Value_AUTO = 0, + HyperparameterScalingType_Value_LINEAR = 1, + HyperparameterScalingType_Value_LOGARITHMIC = 2, + HyperparameterScalingType_Value_REVERSELOGARITHMIC = 3, + HyperparameterScalingType_Value_HyperparameterScalingType_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + HyperparameterScalingType_Value_HyperparameterScalingType_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool HyperparameterScalingType_Value_IsValid(int value); +const HyperparameterScalingType_Value HyperparameterScalingType_Value_Value_MIN = HyperparameterScalingType_Value_AUTO; +const HyperparameterScalingType_Value HyperparameterScalingType_Value_Value_MAX = HyperparameterScalingType_Value_REVERSELOGARITHMIC; +const int HyperparameterScalingType_Value_Value_ARRAYSIZE = HyperparameterScalingType_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* HyperparameterScalingType_Value_descriptor(); +inline const ::std::string& HyperparameterScalingType_Value_Name(HyperparameterScalingType_Value value) { + return ::google::protobuf::internal::NameOfEnum( + HyperparameterScalingType_Value_descriptor(), value); +} +inline bool HyperparameterScalingType_Value_Parse( + const ::std::string& name, HyperparameterScalingType_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + HyperparameterScalingType_Value_descriptor(), name, value); +} +// =================================================================== + +class HyperparameterScalingType final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.HyperparameterScalingType) */ { + public: + HyperparameterScalingType(); + virtual ~HyperparameterScalingType(); + + HyperparameterScalingType(const HyperparameterScalingType& from); + + inline HyperparameterScalingType& operator=(const HyperparameterScalingType& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + HyperparameterScalingType(HyperparameterScalingType&& from) noexcept + : HyperparameterScalingType() { + *this = ::std::move(from); + } + + inline HyperparameterScalingType& operator=(HyperparameterScalingType&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const HyperparameterScalingType& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const HyperparameterScalingType* internal_default_instance() { + return reinterpret_cast( + &_HyperparameterScalingType_default_instance_); + } + static constexpr int kIndexInFileMessages = + 0; + + void Swap(HyperparameterScalingType* other); + friend void swap(HyperparameterScalingType& a, HyperparameterScalingType& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline HyperparameterScalingType* New() const final { + return CreateMaybeMessage(nullptr); + } + + HyperparameterScalingType* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const HyperparameterScalingType& from); + void MergeFrom(const HyperparameterScalingType& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(HyperparameterScalingType* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef HyperparameterScalingType_Value Value; + static const Value AUTO = + HyperparameterScalingType_Value_AUTO; + static const Value LINEAR = + HyperparameterScalingType_Value_LINEAR; + static const Value LOGARITHMIC = + HyperparameterScalingType_Value_LOGARITHMIC; + static const Value REVERSELOGARITHMIC = + HyperparameterScalingType_Value_REVERSELOGARITHMIC; + static inline bool Value_IsValid(int value) { + return HyperparameterScalingType_Value_IsValid(value); + } + static const Value Value_MIN = + HyperparameterScalingType_Value_Value_MIN; + static const Value Value_MAX = + HyperparameterScalingType_Value_Value_MAX; + static const int Value_ARRAYSIZE = + HyperparameterScalingType_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return HyperparameterScalingType_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return HyperparameterScalingType_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return HyperparameterScalingType_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterScalingType) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// ------------------------------------------------------------------- + +class ContinuousParameterRange final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.ContinuousParameterRange) */ { + public: + ContinuousParameterRange(); + virtual ~ContinuousParameterRange(); + + ContinuousParameterRange(const ContinuousParameterRange& from); + + inline ContinuousParameterRange& operator=(const ContinuousParameterRange& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + ContinuousParameterRange(ContinuousParameterRange&& from) noexcept + : ContinuousParameterRange() { + *this = ::std::move(from); + } + + inline ContinuousParameterRange& operator=(ContinuousParameterRange&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const ContinuousParameterRange& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const ContinuousParameterRange* internal_default_instance() { + return reinterpret_cast( + &_ContinuousParameterRange_default_instance_); + } + static constexpr int kIndexInFileMessages = + 1; + + void Swap(ContinuousParameterRange* other); + friend void swap(ContinuousParameterRange& a, ContinuousParameterRange& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline ContinuousParameterRange* New() const final { + return CreateMaybeMessage(nullptr); + } + + ContinuousParameterRange* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const ContinuousParameterRange& from); + void MergeFrom(const ContinuousParameterRange& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(ContinuousParameterRange* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // double max_value = 1; + void clear_max_value(); + static const int kMaxValueFieldNumber = 1; + double max_value() const; + void set_max_value(double value); + + // double min_value = 2; + void clear_min_value(); + static const int kMinValueFieldNumber = 2; + double min_value() const; + void set_min_value(double value); + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + void clear_scaling_type(); + static const int kScalingTypeFieldNumber = 3; + ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value scaling_type() const; + void set_scaling_type(::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ContinuousParameterRange) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + double max_value_; + double min_value_; + int scaling_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// ------------------------------------------------------------------- + +class IntegerParameterRange final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.IntegerParameterRange) */ { + public: + IntegerParameterRange(); + virtual ~IntegerParameterRange(); + + IntegerParameterRange(const IntegerParameterRange& from); + + inline IntegerParameterRange& operator=(const IntegerParameterRange& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + IntegerParameterRange(IntegerParameterRange&& from) noexcept + : IntegerParameterRange() { + *this = ::std::move(from); + } + + inline IntegerParameterRange& operator=(IntegerParameterRange&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const IntegerParameterRange& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const IntegerParameterRange* internal_default_instance() { + return reinterpret_cast( + &_IntegerParameterRange_default_instance_); + } + static constexpr int kIndexInFileMessages = + 2; + + void Swap(IntegerParameterRange* other); + friend void swap(IntegerParameterRange& a, IntegerParameterRange& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline IntegerParameterRange* New() const final { + return CreateMaybeMessage(nullptr); + } + + IntegerParameterRange* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const IntegerParameterRange& from); + void MergeFrom(const IntegerParameterRange& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(IntegerParameterRange* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // int64 max_value = 1; + void clear_max_value(); + static const int kMaxValueFieldNumber = 1; + ::google::protobuf::int64 max_value() const; + void set_max_value(::google::protobuf::int64 value); + + // int64 min_value = 2; + void clear_min_value(); + static const int kMinValueFieldNumber = 2; + ::google::protobuf::int64 min_value() const; + void set_min_value(::google::protobuf::int64 value); + + // .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + void clear_scaling_type(); + static const int kScalingTypeFieldNumber = 3; + ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value scaling_type() const; + void set_scaling_type(::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.IntegerParameterRange) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::int64 max_value_; + ::google::protobuf::int64 min_value_; + int scaling_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// ------------------------------------------------------------------- + +class CategoricalParameterRange final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.CategoricalParameterRange) */ { + public: + CategoricalParameterRange(); + virtual ~CategoricalParameterRange(); + + CategoricalParameterRange(const CategoricalParameterRange& from); + + inline CategoricalParameterRange& operator=(const CategoricalParameterRange& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + CategoricalParameterRange(CategoricalParameterRange&& from) noexcept + : CategoricalParameterRange() { + *this = ::std::move(from); + } + + inline CategoricalParameterRange& operator=(CategoricalParameterRange&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const CategoricalParameterRange& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const CategoricalParameterRange* internal_default_instance() { + return reinterpret_cast( + &_CategoricalParameterRange_default_instance_); + } + static constexpr int kIndexInFileMessages = + 3; + + void Swap(CategoricalParameterRange* other); + friend void swap(CategoricalParameterRange& a, CategoricalParameterRange& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline CategoricalParameterRange* New() const final { + return CreateMaybeMessage(nullptr); + } + + CategoricalParameterRange* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const CategoricalParameterRange& from); + void MergeFrom(const CategoricalParameterRange& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(CategoricalParameterRange* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // repeated string values = 1; + int values_size() const; + void clear_values(); + static const int kValuesFieldNumber = 1; + const ::std::string& values(int index) const; + ::std::string* mutable_values(int index); + void set_values(int index, const ::std::string& value); + #if LANG_CXX11 + void set_values(int index, ::std::string&& value); + #endif + void set_values(int index, const char* value); + void set_values(int index, const char* value, size_t size); + ::std::string* add_values(); + void add_values(const ::std::string& value); + #if LANG_CXX11 + void add_values(::std::string&& value); + #endif + void add_values(const char* value); + void add_values(const char* value, size_t size); + const ::google::protobuf::RepeatedPtrField<::std::string>& values() const; + ::google::protobuf::RepeatedPtrField<::std::string>* mutable_values(); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.CategoricalParameterRange) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::RepeatedPtrField<::std::string> values_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// ------------------------------------------------------------------- + +class ParameterRangeOneOf final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.ParameterRangeOneOf) */ { + public: + ParameterRangeOneOf(); + virtual ~ParameterRangeOneOf(); + + ParameterRangeOneOf(const ParameterRangeOneOf& from); + + inline ParameterRangeOneOf& operator=(const ParameterRangeOneOf& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + ParameterRangeOneOf(ParameterRangeOneOf&& from) noexcept + : ParameterRangeOneOf() { + *this = ::std::move(from); + } + + inline ParameterRangeOneOf& operator=(ParameterRangeOneOf&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const ParameterRangeOneOf& default_instance(); + + enum ParameterRangeTypeCase { + kContinuousParameterRange = 1, + kIntegerParameterRange = 2, + kCategoricalParameterRange = 3, + PARAMETER_RANGE_TYPE_NOT_SET = 0, + }; + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const ParameterRangeOneOf* internal_default_instance() { + return reinterpret_cast( + &_ParameterRangeOneOf_default_instance_); + } + static constexpr int kIndexInFileMessages = + 4; + + void Swap(ParameterRangeOneOf* other); + friend void swap(ParameterRangeOneOf& a, ParameterRangeOneOf& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline ParameterRangeOneOf* New() const final { + return CreateMaybeMessage(nullptr); + } + + ParameterRangeOneOf* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const ParameterRangeOneOf& from); + void MergeFrom(const ParameterRangeOneOf& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(ParameterRangeOneOf* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + bool has_continuous_parameter_range() const; + void clear_continuous_parameter_range(); + static const int kContinuousParameterRangeFieldNumber = 1; + const ::flyteidl::plugins::sagemaker::ContinuousParameterRange& continuous_parameter_range() const; + ::flyteidl::plugins::sagemaker::ContinuousParameterRange* release_continuous_parameter_range(); + ::flyteidl::plugins::sagemaker::ContinuousParameterRange* mutable_continuous_parameter_range(); + void set_allocated_continuous_parameter_range(::flyteidl::plugins::sagemaker::ContinuousParameterRange* continuous_parameter_range); + + // .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + bool has_integer_parameter_range() const; + void clear_integer_parameter_range(); + static const int kIntegerParameterRangeFieldNumber = 2; + const ::flyteidl::plugins::sagemaker::IntegerParameterRange& integer_parameter_range() const; + ::flyteidl::plugins::sagemaker::IntegerParameterRange* release_integer_parameter_range(); + ::flyteidl::plugins::sagemaker::IntegerParameterRange* mutable_integer_parameter_range(); + void set_allocated_integer_parameter_range(::flyteidl::plugins::sagemaker::IntegerParameterRange* integer_parameter_range); + + // .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + bool has_categorical_parameter_range() const; + void clear_categorical_parameter_range(); + static const int kCategoricalParameterRangeFieldNumber = 3; + const ::flyteidl::plugins::sagemaker::CategoricalParameterRange& categorical_parameter_range() const; + ::flyteidl::plugins::sagemaker::CategoricalParameterRange* release_categorical_parameter_range(); + ::flyteidl::plugins::sagemaker::CategoricalParameterRange* mutable_categorical_parameter_range(); + void set_allocated_categorical_parameter_range(::flyteidl::plugins::sagemaker::CategoricalParameterRange* categorical_parameter_range); + + void clear_parameter_range_type(); + ParameterRangeTypeCase parameter_range_type_case() const; + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + private: + class HasBitSetters; + void set_has_continuous_parameter_range(); + void set_has_integer_parameter_range(); + void set_has_categorical_parameter_range(); + + inline bool has_parameter_range_type() const; + inline void clear_has_parameter_range_type(); + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + union ParameterRangeTypeUnion { + ParameterRangeTypeUnion() {} + ::flyteidl::plugins::sagemaker::ContinuousParameterRange* continuous_parameter_range_; + ::flyteidl::plugins::sagemaker::IntegerParameterRange* integer_parameter_range_; + ::flyteidl::plugins::sagemaker::CategoricalParameterRange* categorical_parameter_range_; + } parameter_range_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + ::google::protobuf::uint32 _oneof_case_[1]; + + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// ------------------------------------------------------------------- + +class ParameterRanges_ParameterRangeMapEntry_DoNotUse : public ::google::protobuf::internal::MapEntry { +public: +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +static bool _ParseMap(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + typedef ::google::protobuf::internal::MapEntry SuperType; + ParameterRanges_ParameterRangeMapEntry_DoNotUse(); + ParameterRanges_ParameterRangeMapEntry_DoNotUse(::google::protobuf::Arena* arena); + void MergeFrom(const ParameterRanges_ParameterRangeMapEntry_DoNotUse& other); + static const ParameterRanges_ParameterRangeMapEntry_DoNotUse* internal_default_instance() { return reinterpret_cast(&_ParameterRanges_ParameterRangeMapEntry_DoNotUse_default_instance_); } + void MergeFrom(const ::google::protobuf::Message& other) final; + ::google::protobuf::Metadata GetMetadata() const; +}; + +// ------------------------------------------------------------------- + +class ParameterRanges final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.ParameterRanges) */ { + public: + ParameterRanges(); + virtual ~ParameterRanges(); + + ParameterRanges(const ParameterRanges& from); + + inline ParameterRanges& operator=(const ParameterRanges& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + ParameterRanges(ParameterRanges&& from) noexcept + : ParameterRanges() { + *this = ::std::move(from); + } + + inline ParameterRanges& operator=(ParameterRanges&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const ParameterRanges& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const ParameterRanges* internal_default_instance() { + return reinterpret_cast( + &_ParameterRanges_default_instance_); + } + static constexpr int kIndexInFileMessages = + 6; + + void Swap(ParameterRanges* other); + friend void swap(ParameterRanges& a, ParameterRanges& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline ParameterRanges* New() const final { + return CreateMaybeMessage(nullptr); + } + + ParameterRanges* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const ParameterRanges& from); + void MergeFrom(const ParameterRanges& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(ParameterRanges* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + + // accessors ------------------------------------------------------- + + // map parameter_range_map = 1; + int parameter_range_map_size() const; + void clear_parameter_range_map(); + static const int kParameterRangeMapFieldNumber = 1; + const ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >& + parameter_range_map() const; + ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >* + mutable_parameter_range_map(); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRanges) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::internal::MapField< + ParameterRanges_ParameterRangeMapEntry_DoNotUse, + ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf, + ::google::protobuf::internal::WireFormatLite::TYPE_STRING, + ::google::protobuf::internal::WireFormatLite::TYPE_MESSAGE, + 0 > parameter_range_map_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto; +}; +// =================================================================== + + +// =================================================================== + +#ifdef __GNUC__ + #pragma GCC diagnostic push + #pragma GCC diagnostic ignored "-Wstrict-aliasing" +#endif // __GNUC__ +// HyperparameterScalingType + +// ------------------------------------------------------------------- + +// ContinuousParameterRange + +// double max_value = 1; +inline void ContinuousParameterRange::clear_max_value() { + max_value_ = 0; +} +inline double ContinuousParameterRange::max_value() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ContinuousParameterRange.max_value) + return max_value_; +} +inline void ContinuousParameterRange::set_max_value(double value) { + + max_value_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.ContinuousParameterRange.max_value) +} + +// double min_value = 2; +inline void ContinuousParameterRange::clear_min_value() { + min_value_ = 0; +} +inline double ContinuousParameterRange::min_value() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ContinuousParameterRange.min_value) + return min_value_; +} +inline void ContinuousParameterRange::set_min_value(double value) { + + min_value_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.ContinuousParameterRange.min_value) +} + +// .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; +inline void ContinuousParameterRange::clear_scaling_type() { + scaling_type_ = 0; +} +inline ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value ContinuousParameterRange::scaling_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ContinuousParameterRange.scaling_type) + return static_cast< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value >(scaling_type_); +} +inline void ContinuousParameterRange::set_scaling_type(::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value value) { + + scaling_type_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.ContinuousParameterRange.scaling_type) +} + +// ------------------------------------------------------------------- + +// IntegerParameterRange + +// int64 max_value = 1; +inline void IntegerParameterRange::clear_max_value() { + max_value_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 IntegerParameterRange::max_value() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.IntegerParameterRange.max_value) + return max_value_; +} +inline void IntegerParameterRange::set_max_value(::google::protobuf::int64 value) { + + max_value_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.IntegerParameterRange.max_value) +} + +// int64 min_value = 2; +inline void IntegerParameterRange::clear_min_value() { + min_value_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 IntegerParameterRange::min_value() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.IntegerParameterRange.min_value) + return min_value_; +} +inline void IntegerParameterRange::set_min_value(::google::protobuf::int64 value) { + + min_value_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.IntegerParameterRange.min_value) +} + +// .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; +inline void IntegerParameterRange::clear_scaling_type() { + scaling_type_ = 0; +} +inline ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value IntegerParameterRange::scaling_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.IntegerParameterRange.scaling_type) + return static_cast< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value >(scaling_type_); +} +inline void IntegerParameterRange::set_scaling_type(::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value value) { + + scaling_type_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.IntegerParameterRange.scaling_type) +} + +// ------------------------------------------------------------------- + +// CategoricalParameterRange + +// repeated string values = 1; +inline int CategoricalParameterRange::values_size() const { + return values_.size(); +} +inline void CategoricalParameterRange::clear_values() { + values_.Clear(); +} +inline const ::std::string& CategoricalParameterRange::values(int index) const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + return values_.Get(index); +} +inline ::std::string* CategoricalParameterRange::mutable_values(int index) { + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + return values_.Mutable(index); +} +inline void CategoricalParameterRange::set_values(int index, const ::std::string& value) { + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + values_.Mutable(index)->assign(value); +} +#if LANG_CXX11 +inline void CategoricalParameterRange::set_values(int index, ::std::string&& value) { + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + values_.Mutable(index)->assign(std::move(value)); +} +#endif +inline void CategoricalParameterRange::set_values(int index, const char* value) { + GOOGLE_DCHECK(value != nullptr); + values_.Mutable(index)->assign(value); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +inline void CategoricalParameterRange::set_values(int index, const char* value, size_t size) { + values_.Mutable(index)->assign( + reinterpret_cast(value), size); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +inline ::std::string* CategoricalParameterRange::add_values() { + // @@protoc_insertion_point(field_add_mutable:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + return values_.Add(); +} +inline void CategoricalParameterRange::add_values(const ::std::string& value) { + values_.Add()->assign(value); + // @@protoc_insertion_point(field_add:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +#if LANG_CXX11 +inline void CategoricalParameterRange::add_values(::std::string&& value) { + values_.Add(std::move(value)); + // @@protoc_insertion_point(field_add:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +#endif +inline void CategoricalParameterRange::add_values(const char* value) { + GOOGLE_DCHECK(value != nullptr); + values_.Add()->assign(value); + // @@protoc_insertion_point(field_add_char:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +inline void CategoricalParameterRange::add_values(const char* value, size_t size) { + values_.Add()->assign(reinterpret_cast(value), size); + // @@protoc_insertion_point(field_add_pointer:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) +} +inline const ::google::protobuf::RepeatedPtrField<::std::string>& +CategoricalParameterRange::values() const { + // @@protoc_insertion_point(field_list:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + return values_; +} +inline ::google::protobuf::RepeatedPtrField<::std::string>* +CategoricalParameterRange::mutable_values() { + // @@protoc_insertion_point(field_mutable_list:flyteidl.plugins.sagemaker.CategoricalParameterRange.values) + return &values_; +} + +// ------------------------------------------------------------------- + +// ParameterRangeOneOf + +// .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; +inline bool ParameterRangeOneOf::has_continuous_parameter_range() const { + return parameter_range_type_case() == kContinuousParameterRange; +} +inline void ParameterRangeOneOf::set_has_continuous_parameter_range() { + _oneof_case_[0] = kContinuousParameterRange; +} +inline void ParameterRangeOneOf::clear_continuous_parameter_range() { + if (has_continuous_parameter_range()) { + delete parameter_range_type_.continuous_parameter_range_; + clear_has_parameter_range_type(); + } +} +inline ::flyteidl::plugins::sagemaker::ContinuousParameterRange* ParameterRangeOneOf::release_continuous_parameter_range() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range) + if (has_continuous_parameter_range()) { + clear_has_parameter_range_type(); + ::flyteidl::plugins::sagemaker::ContinuousParameterRange* temp = parameter_range_type_.continuous_parameter_range_; + parameter_range_type_.continuous_parameter_range_ = nullptr; + return temp; + } else { + return nullptr; + } +} +inline const ::flyteidl::plugins::sagemaker::ContinuousParameterRange& ParameterRangeOneOf::continuous_parameter_range() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range) + return has_continuous_parameter_range() + ? *parameter_range_type_.continuous_parameter_range_ + : *reinterpret_cast< ::flyteidl::plugins::sagemaker::ContinuousParameterRange*>(&::flyteidl::plugins::sagemaker::_ContinuousParameterRange_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::ContinuousParameterRange* ParameterRangeOneOf::mutable_continuous_parameter_range() { + if (!has_continuous_parameter_range()) { + clear_parameter_range_type(); + set_has_continuous_parameter_range(); + parameter_range_type_.continuous_parameter_range_ = CreateMaybeMessage< ::flyteidl::plugins::sagemaker::ContinuousParameterRange >( + GetArenaNoVirtual()); + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range) + return parameter_range_type_.continuous_parameter_range_; +} + +// .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; +inline bool ParameterRangeOneOf::has_integer_parameter_range() const { + return parameter_range_type_case() == kIntegerParameterRange; +} +inline void ParameterRangeOneOf::set_has_integer_parameter_range() { + _oneof_case_[0] = kIntegerParameterRange; +} +inline void ParameterRangeOneOf::clear_integer_parameter_range() { + if (has_integer_parameter_range()) { + delete parameter_range_type_.integer_parameter_range_; + clear_has_parameter_range_type(); + } +} +inline ::flyteidl::plugins::sagemaker::IntegerParameterRange* ParameterRangeOneOf::release_integer_parameter_range() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range) + if (has_integer_parameter_range()) { + clear_has_parameter_range_type(); + ::flyteidl::plugins::sagemaker::IntegerParameterRange* temp = parameter_range_type_.integer_parameter_range_; + parameter_range_type_.integer_parameter_range_ = nullptr; + return temp; + } else { + return nullptr; + } +} +inline const ::flyteidl::plugins::sagemaker::IntegerParameterRange& ParameterRangeOneOf::integer_parameter_range() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range) + return has_integer_parameter_range() + ? *parameter_range_type_.integer_parameter_range_ + : *reinterpret_cast< ::flyteidl::plugins::sagemaker::IntegerParameterRange*>(&::flyteidl::plugins::sagemaker::_IntegerParameterRange_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::IntegerParameterRange* ParameterRangeOneOf::mutable_integer_parameter_range() { + if (!has_integer_parameter_range()) { + clear_parameter_range_type(); + set_has_integer_parameter_range(); + parameter_range_type_.integer_parameter_range_ = CreateMaybeMessage< ::flyteidl::plugins::sagemaker::IntegerParameterRange >( + GetArenaNoVirtual()); + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range) + return parameter_range_type_.integer_parameter_range_; +} + +// .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; +inline bool ParameterRangeOneOf::has_categorical_parameter_range() const { + return parameter_range_type_case() == kCategoricalParameterRange; +} +inline void ParameterRangeOneOf::set_has_categorical_parameter_range() { + _oneof_case_[0] = kCategoricalParameterRange; +} +inline void ParameterRangeOneOf::clear_categorical_parameter_range() { + if (has_categorical_parameter_range()) { + delete parameter_range_type_.categorical_parameter_range_; + clear_has_parameter_range_type(); + } +} +inline ::flyteidl::plugins::sagemaker::CategoricalParameterRange* ParameterRangeOneOf::release_categorical_parameter_range() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range) + if (has_categorical_parameter_range()) { + clear_has_parameter_range_type(); + ::flyteidl::plugins::sagemaker::CategoricalParameterRange* temp = parameter_range_type_.categorical_parameter_range_; + parameter_range_type_.categorical_parameter_range_ = nullptr; + return temp; + } else { + return nullptr; + } +} +inline const ::flyteidl::plugins::sagemaker::CategoricalParameterRange& ParameterRangeOneOf::categorical_parameter_range() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range) + return has_categorical_parameter_range() + ? *parameter_range_type_.categorical_parameter_range_ + : *reinterpret_cast< ::flyteidl::plugins::sagemaker::CategoricalParameterRange*>(&::flyteidl::plugins::sagemaker::_CategoricalParameterRange_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::CategoricalParameterRange* ParameterRangeOneOf::mutable_categorical_parameter_range() { + if (!has_categorical_parameter_range()) { + clear_parameter_range_type(); + set_has_categorical_parameter_range(); + parameter_range_type_.categorical_parameter_range_ = CreateMaybeMessage< ::flyteidl::plugins::sagemaker::CategoricalParameterRange >( + GetArenaNoVirtual()); + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range) + return parameter_range_type_.categorical_parameter_range_; +} + +inline bool ParameterRangeOneOf::has_parameter_range_type() const { + return parameter_range_type_case() != PARAMETER_RANGE_TYPE_NOT_SET; +} +inline void ParameterRangeOneOf::clear_has_parameter_range_type() { + _oneof_case_[0] = PARAMETER_RANGE_TYPE_NOT_SET; +} +inline ParameterRangeOneOf::ParameterRangeTypeCase ParameterRangeOneOf::parameter_range_type_case() const { + return ParameterRangeOneOf::ParameterRangeTypeCase(_oneof_case_[0]); +} +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ParameterRanges + +// map parameter_range_map = 1; +inline int ParameterRanges::parameter_range_map_size() const { + return parameter_range_map_.size(); +} +inline void ParameterRanges::clear_parameter_range_map() { + parameter_range_map_.Clear(); +} +inline const ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >& +ParameterRanges::parameter_range_map() const { + // @@protoc_insertion_point(field_map:flyteidl.plugins.sagemaker.ParameterRanges.parameter_range_map) + return parameter_range_map_.GetMap(); +} +inline ::google::protobuf::Map< ::std::string, ::flyteidl::plugins::sagemaker::ParameterRangeOneOf >* +ParameterRanges::mutable_parameter_range_map() { + // @@protoc_insertion_point(field_mutable_map:flyteidl.plugins.sagemaker.ParameterRanges.parameter_range_map) + return parameter_range_map_.MutableMap(); +} + +#ifdef __GNUC__ + #pragma GCC diagnostic pop +#endif // __GNUC__ +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + + +// @@protoc_insertion_point(namespace_scope) + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + +namespace google { +namespace protobuf { + +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value>() { + return ::flyteidl::plugins::sagemaker::HyperparameterScalingType_Value_descriptor(); +} + +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) + +#include +#endif // PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2fparameter_5franges_2eproto diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.cc new file mode 100644 index 000000000..e7b54a6ee --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.cc @@ -0,0 +1,26 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/training_job.proto + +#include "flyteidl/plugins/sagemaker/training_job.pb.h" +#include "flyteidl/plugins/sagemaker/training_job.grpc.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace flyteidl +} // namespace plugins +} // namespace sagemaker + diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.h new file mode 100644 index 000000000..802320ea7 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.grpc.pb.h @@ -0,0 +1,49 @@ +// Generated by the gRPC C++ plugin. +// If you make any local change, they will be lost. +// source: flyteidl/plugins/sagemaker/training_job.proto +#ifndef GRPC_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto__INCLUDED +#define GRPC_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto__INCLUDED + +#include "flyteidl/plugins/sagemaker/training_job.pb.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +namespace grpc_impl { +class Channel; +class CompletionQueue; +class ServerCompletionQueue; +} // namespace grpc_impl + +namespace grpc { +namespace experimental { +template +class MessageAllocator; +} // namespace experimental +} // namespace grpc_impl + +namespace grpc { +class ServerContext; +} // namespace grpc + +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + + +#endif // GRPC_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto__INCLUDED diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.cc b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.cc new file mode 100644 index 000000000..7f846625f --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.cc @@ -0,0 +1,2628 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/training_job.proto + +#include "flyteidl/plugins/sagemaker/training_job.pb.h" + +#include + +#include +#include +#include +#include +#include +#include +#include +#include +// @@protoc_insertion_point(includes) +#include + +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto ::google::protobuf::internal::SCCInfo<0> scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +extern PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto ::google::protobuf::internal::SCCInfo<1> scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class InputModeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _InputMode_default_instance_; +class AlgorithmNameDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _AlgorithmName_default_instance_; +class InputContentTypeDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _InputContentType_default_instance_; +class MetricDefinitionDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _MetricDefinition_default_instance_; +class AlgorithmSpecificationDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _AlgorithmSpecification_default_instance_; +class TrainingJobResourceConfigDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _TrainingJobResourceConfig_default_instance_; +class TrainingJobDefaultTypeInternal { + public: + ::google::protobuf::internal::ExplicitlyConstructed _instance; +} _TrainingJob_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +static void InitDefaultsInputMode_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_InputMode_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::InputMode(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::InputMode::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_InputMode_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsInputMode_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, {}}; + +static void InitDefaultsAlgorithmName_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_AlgorithmName_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::AlgorithmName(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::AlgorithmName::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_AlgorithmName_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsAlgorithmName_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, {}}; + +static void InitDefaultsInputContentType_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_InputContentType_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::InputContentType(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::InputContentType::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_InputContentType_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsInputContentType_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, {}}; + +static void InitDefaultsMetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_MetricDefinition_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::MetricDefinition(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::MetricDefinition::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsMetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, {}}; + +static void InitDefaultsAlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_AlgorithmSpecification_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::AlgorithmSpecification(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::AlgorithmSpecification::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<1> scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 1, InitDefaultsAlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, { + &scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base,}}; + +static void InitDefaultsTrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_TrainingJobResourceConfig_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<0> scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 0, InitDefaultsTrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, {}}; + +static void InitDefaultsTrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + GOOGLE_PROTOBUF_VERIFY_VERSION; + + { + void* ptr = &::flyteidl::plugins::sagemaker::_TrainingJob_default_instance_; + new (ptr) ::flyteidl::plugins::sagemaker::TrainingJob(); + ::google::protobuf::internal::OnShutdownDestroyMessage(ptr); + } + ::flyteidl::plugins::sagemaker::TrainingJob::InitAsDefaultInstance(); +} + +::google::protobuf::internal::SCCInfo<2> scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = + {{ATOMIC_VAR_INIT(::google::protobuf::internal::SCCInfoBase::kUninitialized), 2, InitDefaultsTrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto}, { + &scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base, + &scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base,}}; + +void InitDefaults_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + ::google::protobuf::internal::InitSCC(&scc_info_InputMode_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_AlgorithmName_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_InputContentType_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::google::protobuf::internal::InitSCC(&scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); +} + +::google::protobuf::Metadata file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[7]; +const ::google::protobuf::EnumDescriptor* file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[3]; +constexpr ::google::protobuf::ServiceDescriptor const** file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = nullptr; + +const ::google::protobuf::uint32 TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto::offsets[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::InputMode, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmName, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::InputContentType, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::MetricDefinition, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::MetricDefinition, name_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::MetricDefinition, regex_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, input_mode_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, algorithm_name_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, algorithm_version_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, metric_definitions_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::AlgorithmSpecification, input_content_type_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig, instance_count_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig, instance_type_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig, volume_size_in_gb_), + ~0u, // no _has_bits_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJob, _internal_metadata_), + ~0u, // no _extensions_ + ~0u, // no _oneof_case_ + ~0u, // no _weak_field_map_ + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJob, algorithm_specification_), + PROTOBUF_FIELD_OFFSET(::flyteidl::plugins::sagemaker::TrainingJob, training_job_resource_config_), +}; +static const ::google::protobuf::internal::MigrationSchema schemas[] PROTOBUF_SECTION_VARIABLE(protodesc_cold) = { + { 0, -1, sizeof(::flyteidl::plugins::sagemaker::InputMode)}, + { 5, -1, sizeof(::flyteidl::plugins::sagemaker::AlgorithmName)}, + { 10, -1, sizeof(::flyteidl::plugins::sagemaker::InputContentType)}, + { 15, -1, sizeof(::flyteidl::plugins::sagemaker::MetricDefinition)}, + { 22, -1, sizeof(::flyteidl::plugins::sagemaker::AlgorithmSpecification)}, + { 32, -1, sizeof(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig)}, + { 40, -1, sizeof(::flyteidl::plugins::sagemaker::TrainingJob)}, +}; + +static ::google::protobuf::Message const * const file_default_instances[] = { + reinterpret_cast(&::flyteidl::plugins::sagemaker::_InputMode_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_AlgorithmName_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_InputContentType_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_MetricDefinition_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_AlgorithmSpecification_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_TrainingJobResourceConfig_default_instance_), + reinterpret_cast(&::flyteidl::plugins::sagemaker::_TrainingJob_default_instance_), +}; + +::google::protobuf::internal::AssignDescriptorsTable assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = { + {}, AddDescriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, "flyteidl/plugins/sagemaker/training_job.proto", schemas, + file_default_instances, TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto::offsets, + file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, 7, file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, file_level_service_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, +}; + +const char descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[] = + "\n-flyteidl/plugins/sagemaker/training_jo" + "b.proto\022\032flyteidl.plugins.sagemaker\032\036goo" + "gle/protobuf/duration.proto\"(\n\tInputMode" + "\"\033\n\005Value\022\010\n\004FILE\020\000\022\010\n\004PIPE\020\001\"1\n\rAlgorit" + "hmName\" \n\005Value\022\n\n\006CUSTOM\020\000\022\013\n\007XGBOOST\020\001" + "\")\n\020InputContentType\"\025\n\005Value\022\014\n\010TEXT_CS" + "V\020\000\"/\n\020MetricDefinition\022\014\n\004name\030\001 \001(\t\022\r\n" + "\005regex\030\002 \001(\t\"\327\002\n\026AlgorithmSpecification\022" + "\?\n\ninput_mode\030\001 \001(\0162+.flyteidl.plugins.s" + "agemaker.InputMode.Value\022G\n\016algorithm_na" + "me\030\002 \001(\0162/.flyteidl.plugins.sagemaker.Al" + "gorithmName.Value\022\031\n\021algorithm_version\030\003" + " \001(\t\022H\n\022metric_definitions\030\004 \003(\0132,.flyte" + "idl.plugins.sagemaker.MetricDefinition\022N" + "\n\022input_content_type\030\005 \001(\01622.flyteidl.pl" + "ugins.sagemaker.InputContentType.Value\"e" + "\n\031TrainingJobResourceConfig\022\026\n\016instance_" + "count\030\001 \001(\003\022\025\n\rinstance_type\030\002 \001(\t\022\031\n\021vo" + "lume_size_in_gb\030\003 \001(\003\"\277\001\n\013TrainingJob\022S\n" + "\027algorithm_specification\030\001 \001(\01322.flyteid" + "l.plugins.sagemaker.AlgorithmSpecificati" + "on\022[\n\034training_job_resource_config\030\002 \001(\013" + "25.flyteidl.plugins.sagemaker.TrainingJo" + "bResourceConfigB5Z3github.com/lyft/flyte" + "idl/gen/pb-go/flyteidl/pluginsb\006proto3" + ; +::google::protobuf::internal::DescriptorTable descriptor_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = { + false, InitDefaults_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, + descriptor_table_protodef_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, + "flyteidl/plugins/sagemaker/training_job.proto", &assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, 998, +}; + +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto() { + static constexpr ::google::protobuf::internal::InitFunc deps[1] = + { + ::AddDescriptors_google_2fprotobuf_2fduration_2eproto, + }; + ::google::protobuf::internal::AddDescriptors(&descriptor_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto, deps, 1); +} + +// Force running AddDescriptors() at dynamic initialization time. +static bool dynamic_init_dummy_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto = []() { AddDescriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto(); return true; }(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +const ::google::protobuf::EnumDescriptor* InputMode_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[0]; +} +bool InputMode_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const InputMode_Value InputMode::FILE; +const InputMode_Value InputMode::PIPE; +const InputMode_Value InputMode::Value_MIN; +const InputMode_Value InputMode::Value_MAX; +const int InputMode::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 +const ::google::protobuf::EnumDescriptor* AlgorithmName_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[1]; +} +bool AlgorithmName_Value_IsValid(int value) { + switch (value) { + case 0: + case 1: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const AlgorithmName_Value AlgorithmName::CUSTOM; +const AlgorithmName_Value AlgorithmName::XGBOOST; +const AlgorithmName_Value AlgorithmName::Value_MIN; +const AlgorithmName_Value AlgorithmName::Value_MAX; +const int AlgorithmName::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 +const ::google::protobuf::EnumDescriptor* InputContentType_Value_descriptor() { + ::google::protobuf::internal::AssignDescriptors(&assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return file_level_enum_descriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[2]; +} +bool InputContentType_Value_IsValid(int value) { + switch (value) { + case 0: + return true; + default: + return false; + } +} + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const InputContentType_Value InputContentType::TEXT_CSV; +const InputContentType_Value InputContentType::Value_MIN; +const InputContentType_Value InputContentType::Value_MAX; +const int InputContentType::Value_ARRAYSIZE; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +// =================================================================== + +void InputMode::InitAsDefaultInstance() { +} +class InputMode::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +InputMode::InputMode() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.InputMode) +} +InputMode::InputMode(const InputMode& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.InputMode) +} + +void InputMode::SharedCtor() { +} + +InputMode::~InputMode() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.InputMode) + SharedDtor(); +} + +void InputMode::SharedDtor() { +} + +void InputMode::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const InputMode& InputMode::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_InputMode_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void InputMode::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.InputMode) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* InputMode::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool InputMode::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.InputMode) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.InputMode) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.InputMode) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void InputMode::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.InputMode) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.InputMode) +} + +::google::protobuf::uint8* InputMode::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.InputMode) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.InputMode) + return target; +} + +size_t InputMode::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.InputMode) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void InputMode::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.InputMode) + GOOGLE_DCHECK_NE(&from, this); + const InputMode* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.InputMode) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.InputMode) + MergeFrom(*source); + } +} + +void InputMode::MergeFrom(const InputMode& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.InputMode) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void InputMode::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.InputMode) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void InputMode::CopyFrom(const InputMode& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.InputMode) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool InputMode::IsInitialized() const { + return true; +} + +void InputMode::Swap(InputMode* other) { + if (other == this) return; + InternalSwap(other); +} +void InputMode::InternalSwap(InputMode* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata InputMode::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void AlgorithmName::InitAsDefaultInstance() { +} +class AlgorithmName::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +AlgorithmName::AlgorithmName() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.AlgorithmName) +} +AlgorithmName::AlgorithmName(const AlgorithmName& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.AlgorithmName) +} + +void AlgorithmName::SharedCtor() { +} + +AlgorithmName::~AlgorithmName() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.AlgorithmName) + SharedDtor(); +} + +void AlgorithmName::SharedDtor() { +} + +void AlgorithmName::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const AlgorithmName& AlgorithmName::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_AlgorithmName_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void AlgorithmName::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.AlgorithmName) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* AlgorithmName::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool AlgorithmName::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.AlgorithmName) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.AlgorithmName) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.AlgorithmName) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void AlgorithmName::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.AlgorithmName) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.AlgorithmName) +} + +::google::protobuf::uint8* AlgorithmName::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.AlgorithmName) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.AlgorithmName) + return target; +} + +size_t AlgorithmName::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.AlgorithmName) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void AlgorithmName::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.AlgorithmName) + GOOGLE_DCHECK_NE(&from, this); + const AlgorithmName* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.AlgorithmName) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.AlgorithmName) + MergeFrom(*source); + } +} + +void AlgorithmName::MergeFrom(const AlgorithmName& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.AlgorithmName) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void AlgorithmName::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.AlgorithmName) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void AlgorithmName::CopyFrom(const AlgorithmName& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.AlgorithmName) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool AlgorithmName::IsInitialized() const { + return true; +} + +void AlgorithmName::Swap(AlgorithmName* other) { + if (other == this) return; + InternalSwap(other); +} +void AlgorithmName::InternalSwap(AlgorithmName* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata AlgorithmName::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void InputContentType::InitAsDefaultInstance() { +} +class InputContentType::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +InputContentType::InputContentType() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.InputContentType) +} +InputContentType::InputContentType(const InputContentType& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.InputContentType) +} + +void InputContentType::SharedCtor() { +} + +InputContentType::~InputContentType() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.InputContentType) + SharedDtor(); +} + +void InputContentType::SharedDtor() { +} + +void InputContentType::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const InputContentType& InputContentType::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_InputContentType_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void InputContentType::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.InputContentType) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* InputContentType::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + default: { + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool InputContentType::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.InputContentType) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.InputContentType) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.InputContentType) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void InputContentType::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.InputContentType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.InputContentType) +} + +::google::protobuf::uint8* InputContentType::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.InputContentType) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.InputContentType) + return target; +} + +size_t InputContentType::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.InputContentType) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void InputContentType::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.InputContentType) + GOOGLE_DCHECK_NE(&from, this); + const InputContentType* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.InputContentType) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.InputContentType) + MergeFrom(*source); + } +} + +void InputContentType::MergeFrom(const InputContentType& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.InputContentType) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + +} + +void InputContentType::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.InputContentType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void InputContentType::CopyFrom(const InputContentType& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.InputContentType) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool InputContentType::IsInitialized() const { + return true; +} + +void InputContentType::Swap(InputContentType* other) { + if (other == this) return; + InternalSwap(other); +} +void InputContentType::InternalSwap(InputContentType* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); +} + +::google::protobuf::Metadata InputContentType::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void MetricDefinition::InitAsDefaultInstance() { +} +class MetricDefinition::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int MetricDefinition::kNameFieldNumber; +const int MetricDefinition::kRegexFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +MetricDefinition::MetricDefinition() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.MetricDefinition) +} +MetricDefinition::MetricDefinition(const MetricDefinition& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + name_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + if (from.name().size() > 0) { + name_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.name_); + } + regex_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + if (from.regex().size() > 0) { + regex_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.regex_); + } + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.MetricDefinition) +} + +void MetricDefinition::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + name_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + regex_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} + +MetricDefinition::~MetricDefinition() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.MetricDefinition) + SharedDtor(); +} + +void MetricDefinition::SharedDtor() { + name_.DestroyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + regex_.DestroyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} + +void MetricDefinition::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const MetricDefinition& MetricDefinition::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_MetricDefinition_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void MetricDefinition::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.MetricDefinition) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + name_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + regex_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* MetricDefinition::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // string name = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.MetricDefinition.name"); + object = msg->mutable_name(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + break; + } + // string regex = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 18) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.MetricDefinition.regex"); + object = msg->mutable_regex(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +string_till_end: + static_cast<::std::string*>(object)->clear(); + static_cast<::std::string*>(object)->reserve(size); + goto len_delim_till_end; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool MetricDefinition::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.MetricDefinition) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // string name = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->mutable_name())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->name().data(), static_cast(this->name().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.MetricDefinition.name")); + } else { + goto handle_unusual; + } + break; + } + + // string regex = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (18 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->mutable_regex())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->regex().data(), static_cast(this->regex().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.MetricDefinition.regex")); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.MetricDefinition) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.MetricDefinition) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void MetricDefinition::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.MetricDefinition) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // string name = 1; + if (this->name().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->name().data(), static_cast(this->name().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.MetricDefinition.name"); + ::google::protobuf::internal::WireFormatLite::WriteStringMaybeAliased( + 1, this->name(), output); + } + + // string regex = 2; + if (this->regex().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->regex().data(), static_cast(this->regex().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.MetricDefinition.regex"); + ::google::protobuf::internal::WireFormatLite::WriteStringMaybeAliased( + 2, this->regex(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.MetricDefinition) +} + +::google::protobuf::uint8* MetricDefinition::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.MetricDefinition) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // string name = 1; + if (this->name().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->name().data(), static_cast(this->name().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.MetricDefinition.name"); + target = + ::google::protobuf::internal::WireFormatLite::WriteStringToArray( + 1, this->name(), target); + } + + // string regex = 2; + if (this->regex().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->regex().data(), static_cast(this->regex().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.MetricDefinition.regex"); + target = + ::google::protobuf::internal::WireFormatLite::WriteStringToArray( + 2, this->regex(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.MetricDefinition) + return target; +} + +size_t MetricDefinition::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.MetricDefinition) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // string name = 1; + if (this->name().size() > 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::StringSize( + this->name()); + } + + // string regex = 2; + if (this->regex().size() > 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::StringSize( + this->regex()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void MetricDefinition::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.MetricDefinition) + GOOGLE_DCHECK_NE(&from, this); + const MetricDefinition* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.MetricDefinition) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.MetricDefinition) + MergeFrom(*source); + } +} + +void MetricDefinition::MergeFrom(const MetricDefinition& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.MetricDefinition) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.name().size() > 0) { + + name_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.name_); + } + if (from.regex().size() > 0) { + + regex_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.regex_); + } +} + +void MetricDefinition::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.MetricDefinition) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void MetricDefinition::CopyFrom(const MetricDefinition& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.MetricDefinition) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool MetricDefinition::IsInitialized() const { + return true; +} + +void MetricDefinition::Swap(MetricDefinition* other) { + if (other == this) return; + InternalSwap(other); +} +void MetricDefinition::InternalSwap(MetricDefinition* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + name_.Swap(&other->name_, &::google::protobuf::internal::GetEmptyStringAlreadyInited(), + GetArenaNoVirtual()); + regex_.Swap(&other->regex_, &::google::protobuf::internal::GetEmptyStringAlreadyInited(), + GetArenaNoVirtual()); +} + +::google::protobuf::Metadata MetricDefinition::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void AlgorithmSpecification::InitAsDefaultInstance() { +} +class AlgorithmSpecification::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int AlgorithmSpecification::kInputModeFieldNumber; +const int AlgorithmSpecification::kAlgorithmNameFieldNumber; +const int AlgorithmSpecification::kAlgorithmVersionFieldNumber; +const int AlgorithmSpecification::kMetricDefinitionsFieldNumber; +const int AlgorithmSpecification::kInputContentTypeFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +AlgorithmSpecification::AlgorithmSpecification() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.AlgorithmSpecification) +} +AlgorithmSpecification::AlgorithmSpecification(const AlgorithmSpecification& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr), + metric_definitions_(from.metric_definitions_) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + algorithm_version_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + if (from.algorithm_version().size() > 0) { + algorithm_version_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.algorithm_version_); + } + ::memcpy(&input_mode_, &from.input_mode_, + static_cast(reinterpret_cast(&input_content_type_) - + reinterpret_cast(&input_mode_)) + sizeof(input_content_type_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.AlgorithmSpecification) +} + +void AlgorithmSpecification::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + algorithm_version_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + ::memset(&input_mode_, 0, static_cast( + reinterpret_cast(&input_content_type_) - + reinterpret_cast(&input_mode_)) + sizeof(input_content_type_)); +} + +AlgorithmSpecification::~AlgorithmSpecification() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.AlgorithmSpecification) + SharedDtor(); +} + +void AlgorithmSpecification::SharedDtor() { + algorithm_version_.DestroyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} + +void AlgorithmSpecification::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const AlgorithmSpecification& AlgorithmSpecification::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_AlgorithmSpecification_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void AlgorithmSpecification::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + metric_definitions_.Clear(); + algorithm_version_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + ::memset(&input_mode_, 0, static_cast( + reinterpret_cast(&input_content_type_) - + reinterpret_cast(&input_mode_)) + sizeof(input_content_type_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* AlgorithmSpecification::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 8) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_input_mode(static_cast<::flyteidl::plugins::sagemaker::InputMode_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 16) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_algorithm_name(static_cast<::flyteidl::plugins::sagemaker::AlgorithmName_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // string algorithm_version = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 26) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version"); + object = msg->mutable_algorithm_version(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + break; + } + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + case 4: { + if (static_cast<::google::protobuf::uint8>(tag) != 34) goto handle_unusual; + do { + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::MetricDefinition::_InternalParse; + object = msg->add_metric_definitions(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + if (ptr >= end) break; + } while ((::google::protobuf::io::UnalignedLoad<::google::protobuf::uint64>(ptr) & 255) == 34 && (ptr += 1)); + break; + } + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + case 5: { + if (static_cast<::google::protobuf::uint8>(tag) != 40) goto handle_unusual; + ::google::protobuf::uint64 val = ::google::protobuf::internal::ReadVarint(&ptr); + msg->set_input_content_type(static_cast<::flyteidl::plugins::sagemaker::InputContentType_Value>(val)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +string_till_end: + static_cast<::std::string*>(object)->clear(); + static_cast<::std::string*>(object)->reserve(size); + goto len_delim_till_end; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool AlgorithmSpecification::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (8 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_input_mode(static_cast< ::flyteidl::plugins::sagemaker::InputMode_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (16 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_algorithm_name(static_cast< ::flyteidl::plugins::sagemaker::AlgorithmName_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + // string algorithm_version = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (26 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->mutable_algorithm_version())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->algorithm_version().data(), static_cast(this->algorithm_version().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version")); + } else { + goto handle_unusual; + } + break; + } + + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + case 4: { + if (static_cast< ::google::protobuf::uint8>(tag) == (34 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, add_metric_definitions())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + case 5: { + if (static_cast< ::google::protobuf::uint8>(tag) == (40 & 0xFF)) { + int value = 0; + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + int, ::google::protobuf::internal::WireFormatLite::TYPE_ENUM>( + input, &value))); + set_input_content_type(static_cast< ::flyteidl::plugins::sagemaker::InputContentType_Value >(value)); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.AlgorithmSpecification) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.AlgorithmSpecification) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void AlgorithmSpecification::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + if (this->input_mode() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 1, this->input_mode(), output); + } + + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + if (this->algorithm_name() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 2, this->algorithm_name(), output); + } + + // string algorithm_version = 3; + if (this->algorithm_version().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->algorithm_version().data(), static_cast(this->algorithm_version().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version"); + ::google::protobuf::internal::WireFormatLite::WriteStringMaybeAliased( + 3, this->algorithm_version(), output); + } + + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + for (unsigned int i = 0, + n = static_cast(this->metric_definitions_size()); i < n; i++) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 4, + this->metric_definitions(static_cast(i)), + output); + } + + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + if (this->input_content_type() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteEnum( + 5, this->input_content_type(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.AlgorithmSpecification) +} + +::google::protobuf::uint8* AlgorithmSpecification::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + if (this->input_mode() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 1, this->input_mode(), target); + } + + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + if (this->algorithm_name() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 2, this->algorithm_name(), target); + } + + // string algorithm_version = 3; + if (this->algorithm_version().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->algorithm_version().data(), static_cast(this->algorithm_version().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version"); + target = + ::google::protobuf::internal::WireFormatLite::WriteStringToArray( + 3, this->algorithm_version(), target); + } + + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + for (unsigned int i = 0, + n = static_cast(this->metric_definitions_size()); i < n; i++) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 4, this->metric_definitions(static_cast(i)), target); + } + + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + if (this->input_content_type() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteEnumToArray( + 5, this->input_content_type(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.AlgorithmSpecification) + return target; +} + +size_t AlgorithmSpecification::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + { + unsigned int count = static_cast(this->metric_definitions_size()); + total_size += 1UL * count; + for (unsigned int i = 0; i < count; i++) { + total_size += + ::google::protobuf::internal::WireFormatLite::MessageSize( + this->metric_definitions(static_cast(i))); + } + } + + // string algorithm_version = 3; + if (this->algorithm_version().size() > 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::StringSize( + this->algorithm_version()); + } + + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + if (this->input_mode() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->input_mode()); + } + + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + if (this->algorithm_name() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->algorithm_name()); + } + + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + if (this->input_content_type() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::EnumSize(this->input_content_type()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void AlgorithmSpecification::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + GOOGLE_DCHECK_NE(&from, this); + const AlgorithmSpecification* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.AlgorithmSpecification) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.AlgorithmSpecification) + MergeFrom(*source); + } +} + +void AlgorithmSpecification::MergeFrom(const AlgorithmSpecification& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + metric_definitions_.MergeFrom(from.metric_definitions_); + if (from.algorithm_version().size() > 0) { + + algorithm_version_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.algorithm_version_); + } + if (from.input_mode() != 0) { + set_input_mode(from.input_mode()); + } + if (from.algorithm_name() != 0) { + set_algorithm_name(from.algorithm_name()); + } + if (from.input_content_type() != 0) { + set_input_content_type(from.input_content_type()); + } +} + +void AlgorithmSpecification::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void AlgorithmSpecification::CopyFrom(const AlgorithmSpecification& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.AlgorithmSpecification) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool AlgorithmSpecification::IsInitialized() const { + return true; +} + +void AlgorithmSpecification::Swap(AlgorithmSpecification* other) { + if (other == this) return; + InternalSwap(other); +} +void AlgorithmSpecification::InternalSwap(AlgorithmSpecification* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + CastToBase(&metric_definitions_)->InternalSwap(CastToBase(&other->metric_definitions_)); + algorithm_version_.Swap(&other->algorithm_version_, &::google::protobuf::internal::GetEmptyStringAlreadyInited(), + GetArenaNoVirtual()); + swap(input_mode_, other->input_mode_); + swap(algorithm_name_, other->algorithm_name_); + swap(input_content_type_, other->input_content_type_); +} + +::google::protobuf::Metadata AlgorithmSpecification::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void TrainingJobResourceConfig::InitAsDefaultInstance() { +} +class TrainingJobResourceConfig::HasBitSetters { + public: +}; + +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int TrainingJobResourceConfig::kInstanceCountFieldNumber; +const int TrainingJobResourceConfig::kInstanceTypeFieldNumber; +const int TrainingJobResourceConfig::kVolumeSizeInGbFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +TrainingJobResourceConfig::TrainingJobResourceConfig() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) +} +TrainingJobResourceConfig::TrainingJobResourceConfig(const TrainingJobResourceConfig& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + instance_type_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + if (from.instance_type().size() > 0) { + instance_type_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.instance_type_); + } + ::memcpy(&instance_count_, &from.instance_count_, + static_cast(reinterpret_cast(&volume_size_in_gb_) - + reinterpret_cast(&instance_count_)) + sizeof(volume_size_in_gb_)); + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) +} + +void TrainingJobResourceConfig::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + instance_type_.UnsafeSetDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + ::memset(&instance_count_, 0, static_cast( + reinterpret_cast(&volume_size_in_gb_) - + reinterpret_cast(&instance_count_)) + sizeof(volume_size_in_gb_)); +} + +TrainingJobResourceConfig::~TrainingJobResourceConfig() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + SharedDtor(); +} + +void TrainingJobResourceConfig::SharedDtor() { + instance_type_.DestroyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} + +void TrainingJobResourceConfig::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const TrainingJobResourceConfig& TrainingJobResourceConfig::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_TrainingJobResourceConfig_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void TrainingJobResourceConfig::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + instance_type_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); + ::memset(&instance_count_, 0, static_cast( + reinterpret_cast(&volume_size_in_gb_) - + reinterpret_cast(&instance_count_)) + sizeof(volume_size_in_gb_)); + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* TrainingJobResourceConfig::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // int64 instance_count = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 8) goto handle_unusual; + msg->set_instance_count(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + // string instance_type = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 18) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + ctx->extra_parse_data().SetFieldName("flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type"); + object = msg->mutable_instance_type(); + if (size > end - ptr + ::google::protobuf::internal::ParseContext::kSlopBytes) { + parser_till_end = ::google::protobuf::internal::GreedyStringParserUTF8; + goto string_till_end; + } + GOOGLE_PROTOBUF_PARSER_ASSERT(::google::protobuf::internal::StringCheckUTF8(ptr, size, ctx)); + ::google::protobuf::internal::InlineGreedyStringParser(object, ptr, size, ctx); + ptr += size; + break; + } + // int64 volume_size_in_gb = 3; + case 3: { + if (static_cast<::google::protobuf::uint8>(tag) != 24) goto handle_unusual; + msg->set_volume_size_in_gb(::google::protobuf::internal::ReadVarint(&ptr)); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +string_till_end: + static_cast<::std::string*>(object)->clear(); + static_cast<::std::string*>(object)->reserve(size); + goto len_delim_till_end; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool TrainingJobResourceConfig::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // int64 instance_count = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (8 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &instance_count_))); + } else { + goto handle_unusual; + } + break; + } + + // string instance_type = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (18 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadString( + input, this->mutable_instance_type())); + DO_(::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->instance_type().data(), static_cast(this->instance_type().length()), + ::google::protobuf::internal::WireFormatLite::PARSE, + "flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type")); + } else { + goto handle_unusual; + } + break; + } + + // int64 volume_size_in_gb = 3; + case 3: { + if (static_cast< ::google::protobuf::uint8>(tag) == (24 & 0xFF)) { + + DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive< + ::google::protobuf::int64, ::google::protobuf::internal::WireFormatLite::TYPE_INT64>( + input, &volume_size_in_gb_))); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void TrainingJobResourceConfig::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // int64 instance_count = 1; + if (this->instance_count() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(1, this->instance_count(), output); + } + + // string instance_type = 2; + if (this->instance_type().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->instance_type().data(), static_cast(this->instance_type().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type"); + ::google::protobuf::internal::WireFormatLite::WriteStringMaybeAliased( + 2, this->instance_type(), output); + } + + // int64 volume_size_in_gb = 3; + if (this->volume_size_in_gb() != 0) { + ::google::protobuf::internal::WireFormatLite::WriteInt64(3, this->volume_size_in_gb(), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) +} + +::google::protobuf::uint8* TrainingJobResourceConfig::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // int64 instance_count = 1; + if (this->instance_count() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(1, this->instance_count(), target); + } + + // string instance_type = 2; + if (this->instance_type().size() > 0) { + ::google::protobuf::internal::WireFormatLite::VerifyUtf8String( + this->instance_type().data(), static_cast(this->instance_type().length()), + ::google::protobuf::internal::WireFormatLite::SERIALIZE, + "flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type"); + target = + ::google::protobuf::internal::WireFormatLite::WriteStringToArray( + 2, this->instance_type(), target); + } + + // int64 volume_size_in_gb = 3; + if (this->volume_size_in_gb() != 0) { + target = ::google::protobuf::internal::WireFormatLite::WriteInt64ToArray(3, this->volume_size_in_gb(), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + return target; +} + +size_t TrainingJobResourceConfig::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // string instance_type = 2; + if (this->instance_type().size() > 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::StringSize( + this->instance_type()); + } + + // int64 instance_count = 1; + if (this->instance_count() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->instance_count()); + } + + // int64 volume_size_in_gb = 3; + if (this->volume_size_in_gb() != 0) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::Int64Size( + this->volume_size_in_gb()); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void TrainingJobResourceConfig::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + GOOGLE_DCHECK_NE(&from, this); + const TrainingJobResourceConfig* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + MergeFrom(*source); + } +} + +void TrainingJobResourceConfig::MergeFrom(const TrainingJobResourceConfig& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.instance_type().size() > 0) { + + instance_type_.AssignWithDefault(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), from.instance_type_); + } + if (from.instance_count() != 0) { + set_instance_count(from.instance_count()); + } + if (from.volume_size_in_gb() != 0) { + set_volume_size_in_gb(from.volume_size_in_gb()); + } +} + +void TrainingJobResourceConfig::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void TrainingJobResourceConfig::CopyFrom(const TrainingJobResourceConfig& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool TrainingJobResourceConfig::IsInitialized() const { + return true; +} + +void TrainingJobResourceConfig::Swap(TrainingJobResourceConfig* other) { + if (other == this) return; + InternalSwap(other); +} +void TrainingJobResourceConfig::InternalSwap(TrainingJobResourceConfig* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + instance_type_.Swap(&other->instance_type_, &::google::protobuf::internal::GetEmptyStringAlreadyInited(), + GetArenaNoVirtual()); + swap(instance_count_, other->instance_count_); + swap(volume_size_in_gb_, other->volume_size_in_gb_); +} + +::google::protobuf::Metadata TrainingJobResourceConfig::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// =================================================================== + +void TrainingJob::InitAsDefaultInstance() { + ::flyteidl::plugins::sagemaker::_TrainingJob_default_instance_._instance.get_mutable()->algorithm_specification_ = const_cast< ::flyteidl::plugins::sagemaker::AlgorithmSpecification*>( + ::flyteidl::plugins::sagemaker::AlgorithmSpecification::internal_default_instance()); + ::flyteidl::plugins::sagemaker::_TrainingJob_default_instance_._instance.get_mutable()->training_job_resource_config_ = const_cast< ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig*>( + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig::internal_default_instance()); +} +class TrainingJob::HasBitSetters { + public: + static const ::flyteidl::plugins::sagemaker::AlgorithmSpecification& algorithm_specification(const TrainingJob* msg); + static const ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig& training_job_resource_config(const TrainingJob* msg); +}; + +const ::flyteidl::plugins::sagemaker::AlgorithmSpecification& +TrainingJob::HasBitSetters::algorithm_specification(const TrainingJob* msg) { + return *msg->algorithm_specification_; +} +const ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig& +TrainingJob::HasBitSetters::training_job_resource_config(const TrainingJob* msg) { + return *msg->training_job_resource_config_; +} +#if !defined(_MSC_VER) || _MSC_VER >= 1900 +const int TrainingJob::kAlgorithmSpecificationFieldNumber; +const int TrainingJob::kTrainingJobResourceConfigFieldNumber; +#endif // !defined(_MSC_VER) || _MSC_VER >= 1900 + +TrainingJob::TrainingJob() + : ::google::protobuf::Message(), _internal_metadata_(nullptr) { + SharedCtor(); + // @@protoc_insertion_point(constructor:flyteidl.plugins.sagemaker.TrainingJob) +} +TrainingJob::TrainingJob(const TrainingJob& from) + : ::google::protobuf::Message(), + _internal_metadata_(nullptr) { + _internal_metadata_.MergeFrom(from._internal_metadata_); + if (from.has_algorithm_specification()) { + algorithm_specification_ = new ::flyteidl::plugins::sagemaker::AlgorithmSpecification(*from.algorithm_specification_); + } else { + algorithm_specification_ = nullptr; + } + if (from.has_training_job_resource_config()) { + training_job_resource_config_ = new ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig(*from.training_job_resource_config_); + } else { + training_job_resource_config_ = nullptr; + } + // @@protoc_insertion_point(copy_constructor:flyteidl.plugins.sagemaker.TrainingJob) +} + +void TrainingJob::SharedCtor() { + ::google::protobuf::internal::InitSCC( + &scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + ::memset(&algorithm_specification_, 0, static_cast( + reinterpret_cast(&training_job_resource_config_) - + reinterpret_cast(&algorithm_specification_)) + sizeof(training_job_resource_config_)); +} + +TrainingJob::~TrainingJob() { + // @@protoc_insertion_point(destructor:flyteidl.plugins.sagemaker.TrainingJob) + SharedDtor(); +} + +void TrainingJob::SharedDtor() { + if (this != internal_default_instance()) delete algorithm_specification_; + if (this != internal_default_instance()) delete training_job_resource_config_; +} + +void TrainingJob::SetCachedSize(int size) const { + _cached_size_.Set(size); +} +const TrainingJob& TrainingJob::default_instance() { + ::google::protobuf::internal::InitSCC(&::scc_info_TrainingJob_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto.base); + return *internal_default_instance(); +} + + +void TrainingJob::Clear() { +// @@protoc_insertion_point(message_clear_start:flyteidl.plugins.sagemaker.TrainingJob) + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + if (GetArenaNoVirtual() == nullptr && algorithm_specification_ != nullptr) { + delete algorithm_specification_; + } + algorithm_specification_ = nullptr; + if (GetArenaNoVirtual() == nullptr && training_job_resource_config_ != nullptr) { + delete training_job_resource_config_; + } + training_job_resource_config_ = nullptr; + _internal_metadata_.Clear(); +} + +#if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +const char* TrainingJob::_InternalParse(const char* begin, const char* end, void* object, + ::google::protobuf::internal::ParseContext* ctx) { + auto msg = static_cast(object); + ::google::protobuf::int32 size; (void)size; + int depth; (void)depth; + ::google::protobuf::uint32 tag; + ::google::protobuf::internal::ParseFunc parser_till_end; (void)parser_till_end; + auto ptr = begin; + while (ptr < end) { + ptr = ::google::protobuf::io::Parse32(ptr, &tag); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + switch (tag >> 3) { + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + case 1: { + if (static_cast<::google::protobuf::uint8>(tag) != 10) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::AlgorithmSpecification::_InternalParse; + object = msg->mutable_algorithm_specification(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + case 2: { + if (static_cast<::google::protobuf::uint8>(tag) != 18) goto handle_unusual; + ptr = ::google::protobuf::io::ReadSize(ptr, &size); + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr); + parser_till_end = ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig::_InternalParse; + object = msg->mutable_training_job_resource_config(); + if (size > end - ptr) goto len_delim_till_end; + ptr += size; + GOOGLE_PROTOBUF_PARSER_ASSERT(ctx->ParseExactRange( + {parser_till_end, object}, ptr - size, ptr)); + break; + } + default: { + handle_unusual: + if ((tag & 7) == 4 || tag == 0) { + ctx->EndGroup(tag); + return ptr; + } + auto res = UnknownFieldParse(tag, {_InternalParse, msg}, + ptr, end, msg->_internal_metadata_.mutable_unknown_fields(), ctx); + ptr = res.first; + GOOGLE_PROTOBUF_PARSER_ASSERT(ptr != nullptr); + if (res.second) return ptr; + } + } // switch + } // while + return ptr; +len_delim_till_end: + return ctx->StoreAndTailCall(ptr, end, {_InternalParse, msg}, + {parser_till_end, object}, size); +} +#else // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER +bool TrainingJob::MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) { +#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure + ::google::protobuf::uint32 tag; + // @@protoc_insertion_point(parse_start:flyteidl.plugins.sagemaker.TrainingJob) + for (;;) { + ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u); + tag = p.first; + if (!p.second) goto handle_unusual; + switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) { + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + case 1: { + if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_algorithm_specification())); + } else { + goto handle_unusual; + } + break; + } + + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + case 2: { + if (static_cast< ::google::protobuf::uint8>(tag) == (18 & 0xFF)) { + DO_(::google::protobuf::internal::WireFormatLite::ReadMessage( + input, mutable_training_job_resource_config())); + } else { + goto handle_unusual; + } + break; + } + + default: { + handle_unusual: + if (tag == 0) { + goto success; + } + DO_(::google::protobuf::internal::WireFormat::SkipField( + input, tag, _internal_metadata_.mutable_unknown_fields())); + break; + } + } + } +success: + // @@protoc_insertion_point(parse_success:flyteidl.plugins.sagemaker.TrainingJob) + return true; +failure: + // @@protoc_insertion_point(parse_failure:flyteidl.plugins.sagemaker.TrainingJob) + return false; +#undef DO_ +} +#endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + +void TrainingJob::SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const { + // @@protoc_insertion_point(serialize_start:flyteidl.plugins.sagemaker.TrainingJob) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + if (this->has_algorithm_specification()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 1, HasBitSetters::algorithm_specification(this), output); + } + + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + if (this->has_training_job_resource_config()) { + ::google::protobuf::internal::WireFormatLite::WriteMessageMaybeToArray( + 2, HasBitSetters::training_job_resource_config(this), output); + } + + if (_internal_metadata_.have_unknown_fields()) { + ::google::protobuf::internal::WireFormat::SerializeUnknownFields( + _internal_metadata_.unknown_fields(), output); + } + // @@protoc_insertion_point(serialize_end:flyteidl.plugins.sagemaker.TrainingJob) +} + +::google::protobuf::uint8* TrainingJob::InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const { + // @@protoc_insertion_point(serialize_to_array_start:flyteidl.plugins.sagemaker.TrainingJob) + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + if (this->has_algorithm_specification()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 1, HasBitSetters::algorithm_specification(this), target); + } + + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + if (this->has_training_job_resource_config()) { + target = ::google::protobuf::internal::WireFormatLite:: + InternalWriteMessageToArray( + 2, HasBitSetters::training_job_resource_config(this), target); + } + + if (_internal_metadata_.have_unknown_fields()) { + target = ::google::protobuf::internal::WireFormat::SerializeUnknownFieldsToArray( + _internal_metadata_.unknown_fields(), target); + } + // @@protoc_insertion_point(serialize_to_array_end:flyteidl.plugins.sagemaker.TrainingJob) + return target; +} + +size_t TrainingJob::ByteSizeLong() const { +// @@protoc_insertion_point(message_byte_size_start:flyteidl.plugins.sagemaker.TrainingJob) + size_t total_size = 0; + + if (_internal_metadata_.have_unknown_fields()) { + total_size += + ::google::protobuf::internal::WireFormat::ComputeUnknownFieldsSize( + _internal_metadata_.unknown_fields()); + } + ::google::protobuf::uint32 cached_has_bits = 0; + // Prevent compiler warnings about cached_has_bits being unused + (void) cached_has_bits; + + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + if (this->has_algorithm_specification()) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *algorithm_specification_); + } + + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + if (this->has_training_job_resource_config()) { + total_size += 1 + + ::google::protobuf::internal::WireFormatLite::MessageSize( + *training_job_resource_config_); + } + + int cached_size = ::google::protobuf::internal::ToCachedSize(total_size); + SetCachedSize(cached_size); + return total_size; +} + +void TrainingJob::MergeFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_merge_from_start:flyteidl.plugins.sagemaker.TrainingJob) + GOOGLE_DCHECK_NE(&from, this); + const TrainingJob* source = + ::google::protobuf::DynamicCastToGenerated( + &from); + if (source == nullptr) { + // @@protoc_insertion_point(generalized_merge_from_cast_fail:flyteidl.plugins.sagemaker.TrainingJob) + ::google::protobuf::internal::ReflectionOps::Merge(from, this); + } else { + // @@protoc_insertion_point(generalized_merge_from_cast_success:flyteidl.plugins.sagemaker.TrainingJob) + MergeFrom(*source); + } +} + +void TrainingJob::MergeFrom(const TrainingJob& from) { +// @@protoc_insertion_point(class_specific_merge_from_start:flyteidl.plugins.sagemaker.TrainingJob) + GOOGLE_DCHECK_NE(&from, this); + _internal_metadata_.MergeFrom(from._internal_metadata_); + ::google::protobuf::uint32 cached_has_bits = 0; + (void) cached_has_bits; + + if (from.has_algorithm_specification()) { + mutable_algorithm_specification()->::flyteidl::plugins::sagemaker::AlgorithmSpecification::MergeFrom(from.algorithm_specification()); + } + if (from.has_training_job_resource_config()) { + mutable_training_job_resource_config()->::flyteidl::plugins::sagemaker::TrainingJobResourceConfig::MergeFrom(from.training_job_resource_config()); + } +} + +void TrainingJob::CopyFrom(const ::google::protobuf::Message& from) { +// @@protoc_insertion_point(generalized_copy_from_start:flyteidl.plugins.sagemaker.TrainingJob) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +void TrainingJob::CopyFrom(const TrainingJob& from) { +// @@protoc_insertion_point(class_specific_copy_from_start:flyteidl.plugins.sagemaker.TrainingJob) + if (&from == this) return; + Clear(); + MergeFrom(from); +} + +bool TrainingJob::IsInitialized() const { + return true; +} + +void TrainingJob::Swap(TrainingJob* other) { + if (other == this) return; + InternalSwap(other); +} +void TrainingJob::InternalSwap(TrainingJob* other) { + using std::swap; + _internal_metadata_.Swap(&other->_internal_metadata_); + swap(algorithm_specification_, other->algorithm_specification_); + swap(training_job_resource_config_, other->training_job_resource_config_); +} + +::google::protobuf::Metadata TrainingJob::GetMetadata() const { + ::google::protobuf::internal::AssignDescriptors(&::assign_descriptors_table_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto); + return ::file_level_metadata_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto[kIndexInFileMessages]; +} + + +// @@protoc_insertion_point(namespace_scope) +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::InputMode* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::InputMode >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::InputMode >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::AlgorithmName* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::AlgorithmName >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::AlgorithmName >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::InputContentType* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::InputContentType >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::InputContentType >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::MetricDefinition* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::MetricDefinition >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::MetricDefinition >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::AlgorithmSpecification* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::AlgorithmSpecification >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::AlgorithmSpecification >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig >(arena); +} +template<> PROTOBUF_NOINLINE ::flyteidl::plugins::sagemaker::TrainingJob* Arena::CreateMaybeMessage< ::flyteidl::plugins::sagemaker::TrainingJob >(Arena* arena) { + return Arena::CreateInternal< ::flyteidl::plugins::sagemaker::TrainingJob >(arena); +} +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) +#include diff --git a/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.h b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.h new file mode 100644 index 000000000..ec52b16c2 --- /dev/null +++ b/gen/pb-cpp/flyteidl/plugins/sagemaker/training_job.pb.h @@ -0,0 +1,1592 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/training_job.proto + +#ifndef PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto +#define PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto + +#include +#include + +#include +#if PROTOBUF_VERSION < 3007000 +#error This file was generated by a newer version of protoc which is +#error incompatible with your Protocol Buffer headers. Please update +#error your headers. +#endif +#if 3007000 < PROTOBUF_MIN_PROTOC_VERSION +#error This file was generated by an older version of protoc which is +#error incompatible with your Protocol Buffer headers. Please +#error regenerate this file with a newer version of protoc. +#endif + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include // IWYU pragma: export +#include // IWYU pragma: export +#include +#include +#include +// @@protoc_insertion_point(includes) +#include +#define PROTOBUF_INTERNAL_EXPORT_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto + +// Internal implementation detail -- do not use these members. +struct TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto { + static const ::google::protobuf::internal::ParseTableField entries[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::AuxillaryParseTableField aux[] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::ParseTable schema[7] + PROTOBUF_SECTION_VARIABLE(protodesc_cold); + static const ::google::protobuf::internal::FieldMetadata field_metadata[]; + static const ::google::protobuf::internal::SerializationTable serialization_table[]; + static const ::google::protobuf::uint32 offsets[]; +}; +void AddDescriptors_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto(); +namespace flyteidl { +namespace plugins { +namespace sagemaker { +class AlgorithmName; +class AlgorithmNameDefaultTypeInternal; +extern AlgorithmNameDefaultTypeInternal _AlgorithmName_default_instance_; +class AlgorithmSpecification; +class AlgorithmSpecificationDefaultTypeInternal; +extern AlgorithmSpecificationDefaultTypeInternal _AlgorithmSpecification_default_instance_; +class InputContentType; +class InputContentTypeDefaultTypeInternal; +extern InputContentTypeDefaultTypeInternal _InputContentType_default_instance_; +class InputMode; +class InputModeDefaultTypeInternal; +extern InputModeDefaultTypeInternal _InputMode_default_instance_; +class MetricDefinition; +class MetricDefinitionDefaultTypeInternal; +extern MetricDefinitionDefaultTypeInternal _MetricDefinition_default_instance_; +class TrainingJob; +class TrainingJobDefaultTypeInternal; +extern TrainingJobDefaultTypeInternal _TrainingJob_default_instance_; +class TrainingJobResourceConfig; +class TrainingJobResourceConfigDefaultTypeInternal; +extern TrainingJobResourceConfigDefaultTypeInternal _TrainingJobResourceConfig_default_instance_; +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl +namespace google { +namespace protobuf { +template<> ::flyteidl::plugins::sagemaker::AlgorithmName* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::AlgorithmName>(Arena*); +template<> ::flyteidl::plugins::sagemaker::AlgorithmSpecification* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::AlgorithmSpecification>(Arena*); +template<> ::flyteidl::plugins::sagemaker::InputContentType* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::InputContentType>(Arena*); +template<> ::flyteidl::plugins::sagemaker::InputMode* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::InputMode>(Arena*); +template<> ::flyteidl::plugins::sagemaker::MetricDefinition* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::MetricDefinition>(Arena*); +template<> ::flyteidl::plugins::sagemaker::TrainingJob* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::TrainingJob>(Arena*); +template<> ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* Arena::CreateMaybeMessage<::flyteidl::plugins::sagemaker::TrainingJobResourceConfig>(Arena*); +} // namespace protobuf +} // namespace google +namespace flyteidl { +namespace plugins { +namespace sagemaker { + +enum InputMode_Value { + InputMode_Value_FILE = 0, + InputMode_Value_PIPE = 1, + InputMode_Value_InputMode_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + InputMode_Value_InputMode_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool InputMode_Value_IsValid(int value); +const InputMode_Value InputMode_Value_Value_MIN = InputMode_Value_FILE; +const InputMode_Value InputMode_Value_Value_MAX = InputMode_Value_PIPE; +const int InputMode_Value_Value_ARRAYSIZE = InputMode_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* InputMode_Value_descriptor(); +inline const ::std::string& InputMode_Value_Name(InputMode_Value value) { + return ::google::protobuf::internal::NameOfEnum( + InputMode_Value_descriptor(), value); +} +inline bool InputMode_Value_Parse( + const ::std::string& name, InputMode_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + InputMode_Value_descriptor(), name, value); +} +enum AlgorithmName_Value { + AlgorithmName_Value_CUSTOM = 0, + AlgorithmName_Value_XGBOOST = 1, + AlgorithmName_Value_AlgorithmName_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + AlgorithmName_Value_AlgorithmName_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool AlgorithmName_Value_IsValid(int value); +const AlgorithmName_Value AlgorithmName_Value_Value_MIN = AlgorithmName_Value_CUSTOM; +const AlgorithmName_Value AlgorithmName_Value_Value_MAX = AlgorithmName_Value_XGBOOST; +const int AlgorithmName_Value_Value_ARRAYSIZE = AlgorithmName_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* AlgorithmName_Value_descriptor(); +inline const ::std::string& AlgorithmName_Value_Name(AlgorithmName_Value value) { + return ::google::protobuf::internal::NameOfEnum( + AlgorithmName_Value_descriptor(), value); +} +inline bool AlgorithmName_Value_Parse( + const ::std::string& name, AlgorithmName_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + AlgorithmName_Value_descriptor(), name, value); +} +enum InputContentType_Value { + InputContentType_Value_TEXT_CSV = 0, + InputContentType_Value_InputContentType_Value_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::min(), + InputContentType_Value_InputContentType_Value_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<::google::protobuf::int32>::max() +}; +bool InputContentType_Value_IsValid(int value); +const InputContentType_Value InputContentType_Value_Value_MIN = InputContentType_Value_TEXT_CSV; +const InputContentType_Value InputContentType_Value_Value_MAX = InputContentType_Value_TEXT_CSV; +const int InputContentType_Value_Value_ARRAYSIZE = InputContentType_Value_Value_MAX + 1; + +const ::google::protobuf::EnumDescriptor* InputContentType_Value_descriptor(); +inline const ::std::string& InputContentType_Value_Name(InputContentType_Value value) { + return ::google::protobuf::internal::NameOfEnum( + InputContentType_Value_descriptor(), value); +} +inline bool InputContentType_Value_Parse( + const ::std::string& name, InputContentType_Value* value) { + return ::google::protobuf::internal::ParseNamedEnum( + InputContentType_Value_descriptor(), name, value); +} +// =================================================================== + +class InputMode final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.InputMode) */ { + public: + InputMode(); + virtual ~InputMode(); + + InputMode(const InputMode& from); + + inline InputMode& operator=(const InputMode& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + InputMode(InputMode&& from) noexcept + : InputMode() { + *this = ::std::move(from); + } + + inline InputMode& operator=(InputMode&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const InputMode& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const InputMode* internal_default_instance() { + return reinterpret_cast( + &_InputMode_default_instance_); + } + static constexpr int kIndexInFileMessages = + 0; + + void Swap(InputMode* other); + friend void swap(InputMode& a, InputMode& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline InputMode* New() const final { + return CreateMaybeMessage(nullptr); + } + + InputMode* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const InputMode& from); + void MergeFrom(const InputMode& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(InputMode* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef InputMode_Value Value; + static const Value FILE = + InputMode_Value_FILE; + static const Value PIPE = + InputMode_Value_PIPE; + static inline bool Value_IsValid(int value) { + return InputMode_Value_IsValid(value); + } + static const Value Value_MIN = + InputMode_Value_Value_MIN; + static const Value Value_MAX = + InputMode_Value_Value_MAX; + static const int Value_ARRAYSIZE = + InputMode_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return InputMode_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return InputMode_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return InputMode_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputMode) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class AlgorithmName final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.AlgorithmName) */ { + public: + AlgorithmName(); + virtual ~AlgorithmName(); + + AlgorithmName(const AlgorithmName& from); + + inline AlgorithmName& operator=(const AlgorithmName& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + AlgorithmName(AlgorithmName&& from) noexcept + : AlgorithmName() { + *this = ::std::move(from); + } + + inline AlgorithmName& operator=(AlgorithmName&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const AlgorithmName& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const AlgorithmName* internal_default_instance() { + return reinterpret_cast( + &_AlgorithmName_default_instance_); + } + static constexpr int kIndexInFileMessages = + 1; + + void Swap(AlgorithmName* other); + friend void swap(AlgorithmName& a, AlgorithmName& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline AlgorithmName* New() const final { + return CreateMaybeMessage(nullptr); + } + + AlgorithmName* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const AlgorithmName& from); + void MergeFrom(const AlgorithmName& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(AlgorithmName* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef AlgorithmName_Value Value; + static const Value CUSTOM = + AlgorithmName_Value_CUSTOM; + static const Value XGBOOST = + AlgorithmName_Value_XGBOOST; + static inline bool Value_IsValid(int value) { + return AlgorithmName_Value_IsValid(value); + } + static const Value Value_MIN = + AlgorithmName_Value_Value_MIN; + static const Value Value_MAX = + AlgorithmName_Value_Value_MAX; + static const int Value_ARRAYSIZE = + AlgorithmName_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return AlgorithmName_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return AlgorithmName_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return AlgorithmName_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmName) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class InputContentType final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.InputContentType) */ { + public: + InputContentType(); + virtual ~InputContentType(); + + InputContentType(const InputContentType& from); + + inline InputContentType& operator=(const InputContentType& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + InputContentType(InputContentType&& from) noexcept + : InputContentType() { + *this = ::std::move(from); + } + + inline InputContentType& operator=(InputContentType&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const InputContentType& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const InputContentType* internal_default_instance() { + return reinterpret_cast( + &_InputContentType_default_instance_); + } + static constexpr int kIndexInFileMessages = + 2; + + void Swap(InputContentType* other); + friend void swap(InputContentType& a, InputContentType& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline InputContentType* New() const final { + return CreateMaybeMessage(nullptr); + } + + InputContentType* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const InputContentType& from); + void MergeFrom(const InputContentType& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(InputContentType* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + typedef InputContentType_Value Value; + static const Value TEXT_CSV = + InputContentType_Value_TEXT_CSV; + static inline bool Value_IsValid(int value) { + return InputContentType_Value_IsValid(value); + } + static const Value Value_MIN = + InputContentType_Value_Value_MIN; + static const Value Value_MAX = + InputContentType_Value_Value_MAX; + static const int Value_ARRAYSIZE = + InputContentType_Value_Value_ARRAYSIZE; + static inline const ::google::protobuf::EnumDescriptor* + Value_descriptor() { + return InputContentType_Value_descriptor(); + } + static inline const ::std::string& Value_Name(Value value) { + return InputContentType_Value_Name(value); + } + static inline bool Value_Parse(const ::std::string& name, + Value* value) { + return InputContentType_Value_Parse(name, value); + } + + // accessors ------------------------------------------------------- + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputContentType) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class MetricDefinition final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.MetricDefinition) */ { + public: + MetricDefinition(); + virtual ~MetricDefinition(); + + MetricDefinition(const MetricDefinition& from); + + inline MetricDefinition& operator=(const MetricDefinition& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + MetricDefinition(MetricDefinition&& from) noexcept + : MetricDefinition() { + *this = ::std::move(from); + } + + inline MetricDefinition& operator=(MetricDefinition&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const MetricDefinition& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const MetricDefinition* internal_default_instance() { + return reinterpret_cast( + &_MetricDefinition_default_instance_); + } + static constexpr int kIndexInFileMessages = + 3; + + void Swap(MetricDefinition* other); + friend void swap(MetricDefinition& a, MetricDefinition& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline MetricDefinition* New() const final { + return CreateMaybeMessage(nullptr); + } + + MetricDefinition* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const MetricDefinition& from); + void MergeFrom(const MetricDefinition& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(MetricDefinition* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // string name = 1; + void clear_name(); + static const int kNameFieldNumber = 1; + const ::std::string& name() const; + void set_name(const ::std::string& value); + #if LANG_CXX11 + void set_name(::std::string&& value); + #endif + void set_name(const char* value); + void set_name(const char* value, size_t size); + ::std::string* mutable_name(); + ::std::string* release_name(); + void set_allocated_name(::std::string* name); + + // string regex = 2; + void clear_regex(); + static const int kRegexFieldNumber = 2; + const ::std::string& regex() const; + void set_regex(const ::std::string& value); + #if LANG_CXX11 + void set_regex(::std::string&& value); + #endif + void set_regex(const char* value); + void set_regex(const char* value, size_t size); + ::std::string* mutable_regex(); + ::std::string* release_regex(); + void set_allocated_regex(::std::string* regex); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.MetricDefinition) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::internal::ArenaStringPtr name_; + ::google::protobuf::internal::ArenaStringPtr regex_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class AlgorithmSpecification final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.AlgorithmSpecification) */ { + public: + AlgorithmSpecification(); + virtual ~AlgorithmSpecification(); + + AlgorithmSpecification(const AlgorithmSpecification& from); + + inline AlgorithmSpecification& operator=(const AlgorithmSpecification& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + AlgorithmSpecification(AlgorithmSpecification&& from) noexcept + : AlgorithmSpecification() { + *this = ::std::move(from); + } + + inline AlgorithmSpecification& operator=(AlgorithmSpecification&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const AlgorithmSpecification& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const AlgorithmSpecification* internal_default_instance() { + return reinterpret_cast( + &_AlgorithmSpecification_default_instance_); + } + static constexpr int kIndexInFileMessages = + 4; + + void Swap(AlgorithmSpecification* other); + friend void swap(AlgorithmSpecification& a, AlgorithmSpecification& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline AlgorithmSpecification* New() const final { + return CreateMaybeMessage(nullptr); + } + + AlgorithmSpecification* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const AlgorithmSpecification& from); + void MergeFrom(const AlgorithmSpecification& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(AlgorithmSpecification* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + int metric_definitions_size() const; + void clear_metric_definitions(); + static const int kMetricDefinitionsFieldNumber = 4; + ::flyteidl::plugins::sagemaker::MetricDefinition* mutable_metric_definitions(int index); + ::google::protobuf::RepeatedPtrField< ::flyteidl::plugins::sagemaker::MetricDefinition >* + mutable_metric_definitions(); + const ::flyteidl::plugins::sagemaker::MetricDefinition& metric_definitions(int index) const; + ::flyteidl::plugins::sagemaker::MetricDefinition* add_metric_definitions(); + const ::google::protobuf::RepeatedPtrField< ::flyteidl::plugins::sagemaker::MetricDefinition >& + metric_definitions() const; + + // string algorithm_version = 3; + void clear_algorithm_version(); + static const int kAlgorithmVersionFieldNumber = 3; + const ::std::string& algorithm_version() const; + void set_algorithm_version(const ::std::string& value); + #if LANG_CXX11 + void set_algorithm_version(::std::string&& value); + #endif + void set_algorithm_version(const char* value); + void set_algorithm_version(const char* value, size_t size); + ::std::string* mutable_algorithm_version(); + ::std::string* release_algorithm_version(); + void set_allocated_algorithm_version(::std::string* algorithm_version); + + // .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + void clear_input_mode(); + static const int kInputModeFieldNumber = 1; + ::flyteidl::plugins::sagemaker::InputMode_Value input_mode() const; + void set_input_mode(::flyteidl::plugins::sagemaker::InputMode_Value value); + + // .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + void clear_algorithm_name(); + static const int kAlgorithmNameFieldNumber = 2; + ::flyteidl::plugins::sagemaker::AlgorithmName_Value algorithm_name() const; + void set_algorithm_name(::flyteidl::plugins::sagemaker::AlgorithmName_Value value); + + // .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + void clear_input_content_type(); + static const int kInputContentTypeFieldNumber = 5; + ::flyteidl::plugins::sagemaker::InputContentType_Value input_content_type() const; + void set_input_content_type(::flyteidl::plugins::sagemaker::InputContentType_Value value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmSpecification) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::RepeatedPtrField< ::flyteidl::plugins::sagemaker::MetricDefinition > metric_definitions_; + ::google::protobuf::internal::ArenaStringPtr algorithm_version_; + int input_mode_; + int algorithm_name_; + int input_content_type_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class TrainingJobResourceConfig final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) */ { + public: + TrainingJobResourceConfig(); + virtual ~TrainingJobResourceConfig(); + + TrainingJobResourceConfig(const TrainingJobResourceConfig& from); + + inline TrainingJobResourceConfig& operator=(const TrainingJobResourceConfig& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + TrainingJobResourceConfig(TrainingJobResourceConfig&& from) noexcept + : TrainingJobResourceConfig() { + *this = ::std::move(from); + } + + inline TrainingJobResourceConfig& operator=(TrainingJobResourceConfig&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const TrainingJobResourceConfig& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const TrainingJobResourceConfig* internal_default_instance() { + return reinterpret_cast( + &_TrainingJobResourceConfig_default_instance_); + } + static constexpr int kIndexInFileMessages = + 5; + + void Swap(TrainingJobResourceConfig* other); + friend void swap(TrainingJobResourceConfig& a, TrainingJobResourceConfig& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline TrainingJobResourceConfig* New() const final { + return CreateMaybeMessage(nullptr); + } + + TrainingJobResourceConfig* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const TrainingJobResourceConfig& from); + void MergeFrom(const TrainingJobResourceConfig& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(TrainingJobResourceConfig* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // string instance_type = 2; + void clear_instance_type(); + static const int kInstanceTypeFieldNumber = 2; + const ::std::string& instance_type() const; + void set_instance_type(const ::std::string& value); + #if LANG_CXX11 + void set_instance_type(::std::string&& value); + #endif + void set_instance_type(const char* value); + void set_instance_type(const char* value, size_t size); + ::std::string* mutable_instance_type(); + ::std::string* release_instance_type(); + void set_allocated_instance_type(::std::string* instance_type); + + // int64 instance_count = 1; + void clear_instance_count(); + static const int kInstanceCountFieldNumber = 1; + ::google::protobuf::int64 instance_count() const; + void set_instance_count(::google::protobuf::int64 value); + + // int64 volume_size_in_gb = 3; + void clear_volume_size_in_gb(); + static const int kVolumeSizeInGbFieldNumber = 3; + ::google::protobuf::int64 volume_size_in_gb() const; + void set_volume_size_in_gb(::google::protobuf::int64 value); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::google::protobuf::internal::ArenaStringPtr instance_type_; + ::google::protobuf::int64 instance_count_; + ::google::protobuf::int64 volume_size_in_gb_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// ------------------------------------------------------------------- + +class TrainingJob final : + public ::google::protobuf::Message /* @@protoc_insertion_point(class_definition:flyteidl.plugins.sagemaker.TrainingJob) */ { + public: + TrainingJob(); + virtual ~TrainingJob(); + + TrainingJob(const TrainingJob& from); + + inline TrainingJob& operator=(const TrainingJob& from) { + CopyFrom(from); + return *this; + } + #if LANG_CXX11 + TrainingJob(TrainingJob&& from) noexcept + : TrainingJob() { + *this = ::std::move(from); + } + + inline TrainingJob& operator=(TrainingJob&& from) noexcept { + if (GetArenaNoVirtual() == from.GetArenaNoVirtual()) { + if (this != &from) InternalSwap(&from); + } else { + CopyFrom(from); + } + return *this; + } + #endif + static const ::google::protobuf::Descriptor* descriptor() { + return default_instance().GetDescriptor(); + } + static const TrainingJob& default_instance(); + + static void InitAsDefaultInstance(); // FOR INTERNAL USE ONLY + static inline const TrainingJob* internal_default_instance() { + return reinterpret_cast( + &_TrainingJob_default_instance_); + } + static constexpr int kIndexInFileMessages = + 6; + + void Swap(TrainingJob* other); + friend void swap(TrainingJob& a, TrainingJob& b) { + a.Swap(&b); + } + + // implements Message ---------------------------------------------- + + inline TrainingJob* New() const final { + return CreateMaybeMessage(nullptr); + } + + TrainingJob* New(::google::protobuf::Arena* arena) const final { + return CreateMaybeMessage(arena); + } + void CopyFrom(const ::google::protobuf::Message& from) final; + void MergeFrom(const ::google::protobuf::Message& from) final; + void CopyFrom(const TrainingJob& from); + void MergeFrom(const TrainingJob& from); + PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final; + bool IsInitialized() const final; + + size_t ByteSizeLong() const final; + #if GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + static const char* _InternalParse(const char* begin, const char* end, void* object, ::google::protobuf::internal::ParseContext* ctx); + ::google::protobuf::internal::ParseFunc _ParseFunc() const final { return _InternalParse; } + #else + bool MergePartialFromCodedStream( + ::google::protobuf::io::CodedInputStream* input) final; + #endif // GOOGLE_PROTOBUF_ENABLE_EXPERIMENTAL_PARSER + void SerializeWithCachedSizes( + ::google::protobuf::io::CodedOutputStream* output) const final; + ::google::protobuf::uint8* InternalSerializeWithCachedSizesToArray( + ::google::protobuf::uint8* target) const final; + int GetCachedSize() const final { return _cached_size_.Get(); } + + private: + void SharedCtor(); + void SharedDtor(); + void SetCachedSize(int size) const final; + void InternalSwap(TrainingJob* other); + private: + inline ::google::protobuf::Arena* GetArenaNoVirtual() const { + return nullptr; + } + inline void* MaybeArenaPtr() const { + return nullptr; + } + public: + + ::google::protobuf::Metadata GetMetadata() const final; + + // nested types ---------------------------------------------------- + + // accessors ------------------------------------------------------- + + // .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + bool has_algorithm_specification() const; + void clear_algorithm_specification(); + static const int kAlgorithmSpecificationFieldNumber = 1; + const ::flyteidl::plugins::sagemaker::AlgorithmSpecification& algorithm_specification() const; + ::flyteidl::plugins::sagemaker::AlgorithmSpecification* release_algorithm_specification(); + ::flyteidl::plugins::sagemaker::AlgorithmSpecification* mutable_algorithm_specification(); + void set_allocated_algorithm_specification(::flyteidl::plugins::sagemaker::AlgorithmSpecification* algorithm_specification); + + // .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + bool has_training_job_resource_config() const; + void clear_training_job_resource_config(); + static const int kTrainingJobResourceConfigFieldNumber = 2; + const ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig& training_job_resource_config() const; + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* release_training_job_resource_config(); + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* mutable_training_job_resource_config(); + void set_allocated_training_job_resource_config(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* training_job_resource_config); + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJob) + private: + class HasBitSetters; + + ::google::protobuf::internal::InternalMetadataWithArena _internal_metadata_; + ::flyteidl::plugins::sagemaker::AlgorithmSpecification* algorithm_specification_; + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* training_job_resource_config_; + mutable ::google::protobuf::internal::CachedSize _cached_size_; + friend struct ::TableStruct_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto; +}; +// =================================================================== + + +// =================================================================== + +#ifdef __GNUC__ + #pragma GCC diagnostic push + #pragma GCC diagnostic ignored "-Wstrict-aliasing" +#endif // __GNUC__ +// InputMode + +// ------------------------------------------------------------------- + +// AlgorithmName + +// ------------------------------------------------------------------- + +// InputContentType + +// ------------------------------------------------------------------- + +// MetricDefinition + +// string name = 1; +inline void MetricDefinition::clear_name() { + name_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline const ::std::string& MetricDefinition::name() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.MetricDefinition.name) + return name_.GetNoArena(); +} +inline void MetricDefinition::set_name(const ::std::string& value) { + + name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), value); + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.MetricDefinition.name) +} +#if LANG_CXX11 +inline void MetricDefinition::set_name(::std::string&& value) { + + name_.SetNoArena( + &::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::move(value)); + // @@protoc_insertion_point(field_set_rvalue:flyteidl.plugins.sagemaker.MetricDefinition.name) +} +#endif +inline void MetricDefinition::set_name(const char* value) { + GOOGLE_DCHECK(value != nullptr); + + name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::string(value)); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.MetricDefinition.name) +} +inline void MetricDefinition::set_name(const char* value, size_t size) { + + name_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), + ::std::string(reinterpret_cast(value), size)); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.MetricDefinition.name) +} +inline ::std::string* MetricDefinition::mutable_name() { + + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.MetricDefinition.name) + return name_.MutableNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline ::std::string* MetricDefinition::release_name() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.MetricDefinition.name) + + return name_.ReleaseNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline void MetricDefinition::set_allocated_name(::std::string* name) { + if (name != nullptr) { + + } else { + + } + name_.SetAllocatedNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), name); + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.MetricDefinition.name) +} + +// string regex = 2; +inline void MetricDefinition::clear_regex() { + regex_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline const ::std::string& MetricDefinition::regex() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.MetricDefinition.regex) + return regex_.GetNoArena(); +} +inline void MetricDefinition::set_regex(const ::std::string& value) { + + regex_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), value); + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.MetricDefinition.regex) +} +#if LANG_CXX11 +inline void MetricDefinition::set_regex(::std::string&& value) { + + regex_.SetNoArena( + &::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::move(value)); + // @@protoc_insertion_point(field_set_rvalue:flyteidl.plugins.sagemaker.MetricDefinition.regex) +} +#endif +inline void MetricDefinition::set_regex(const char* value) { + GOOGLE_DCHECK(value != nullptr); + + regex_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::string(value)); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.MetricDefinition.regex) +} +inline void MetricDefinition::set_regex(const char* value, size_t size) { + + regex_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), + ::std::string(reinterpret_cast(value), size)); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.MetricDefinition.regex) +} +inline ::std::string* MetricDefinition::mutable_regex() { + + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.MetricDefinition.regex) + return regex_.MutableNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline ::std::string* MetricDefinition::release_regex() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.MetricDefinition.regex) + + return regex_.ReleaseNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline void MetricDefinition::set_allocated_regex(::std::string* regex) { + if (regex != nullptr) { + + } else { + + } + regex_.SetAllocatedNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), regex); + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.MetricDefinition.regex) +} + +// ------------------------------------------------------------------- + +// AlgorithmSpecification + +// .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; +inline void AlgorithmSpecification::clear_input_mode() { + input_mode_ = 0; +} +inline ::flyteidl::plugins::sagemaker::InputMode_Value AlgorithmSpecification::input_mode() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.AlgorithmSpecification.input_mode) + return static_cast< ::flyteidl::plugins::sagemaker::InputMode_Value >(input_mode_); +} +inline void AlgorithmSpecification::set_input_mode(::flyteidl::plugins::sagemaker::InputMode_Value value) { + + input_mode_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.AlgorithmSpecification.input_mode) +} + +// .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; +inline void AlgorithmSpecification::clear_algorithm_name() { + algorithm_name_ = 0; +} +inline ::flyteidl::plugins::sagemaker::AlgorithmName_Value AlgorithmSpecification::algorithm_name() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_name) + return static_cast< ::flyteidl::plugins::sagemaker::AlgorithmName_Value >(algorithm_name_); +} +inline void AlgorithmSpecification::set_algorithm_name(::flyteidl::plugins::sagemaker::AlgorithmName_Value value) { + + algorithm_name_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_name) +} + +// string algorithm_version = 3; +inline void AlgorithmSpecification::clear_algorithm_version() { + algorithm_version_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline const ::std::string& AlgorithmSpecification::algorithm_version() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) + return algorithm_version_.GetNoArena(); +} +inline void AlgorithmSpecification::set_algorithm_version(const ::std::string& value) { + + algorithm_version_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), value); + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) +} +#if LANG_CXX11 +inline void AlgorithmSpecification::set_algorithm_version(::std::string&& value) { + + algorithm_version_.SetNoArena( + &::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::move(value)); + // @@protoc_insertion_point(field_set_rvalue:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) +} +#endif +inline void AlgorithmSpecification::set_algorithm_version(const char* value) { + GOOGLE_DCHECK(value != nullptr); + + algorithm_version_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::string(value)); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) +} +inline void AlgorithmSpecification::set_algorithm_version(const char* value, size_t size) { + + algorithm_version_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), + ::std::string(reinterpret_cast(value), size)); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) +} +inline ::std::string* AlgorithmSpecification::mutable_algorithm_version() { + + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) + return algorithm_version_.MutableNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline ::std::string* AlgorithmSpecification::release_algorithm_version() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) + + return algorithm_version_.ReleaseNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline void AlgorithmSpecification::set_allocated_algorithm_version(::std::string* algorithm_version) { + if (algorithm_version != nullptr) { + + } else { + + } + algorithm_version_.SetAllocatedNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), algorithm_version); + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version) +} + +// repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; +inline int AlgorithmSpecification::metric_definitions_size() const { + return metric_definitions_.size(); +} +inline void AlgorithmSpecification::clear_metric_definitions() { + metric_definitions_.Clear(); +} +inline ::flyteidl::plugins::sagemaker::MetricDefinition* AlgorithmSpecification::mutable_metric_definitions(int index) { + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions) + return metric_definitions_.Mutable(index); +} +inline ::google::protobuf::RepeatedPtrField< ::flyteidl::plugins::sagemaker::MetricDefinition >* +AlgorithmSpecification::mutable_metric_definitions() { + // @@protoc_insertion_point(field_mutable_list:flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions) + return &metric_definitions_; +} +inline const ::flyteidl::plugins::sagemaker::MetricDefinition& AlgorithmSpecification::metric_definitions(int index) const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions) + return metric_definitions_.Get(index); +} +inline ::flyteidl::plugins::sagemaker::MetricDefinition* AlgorithmSpecification::add_metric_definitions() { + // @@protoc_insertion_point(field_add:flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions) + return metric_definitions_.Add(); +} +inline const ::google::protobuf::RepeatedPtrField< ::flyteidl::plugins::sagemaker::MetricDefinition >& +AlgorithmSpecification::metric_definitions() const { + // @@protoc_insertion_point(field_list:flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions) + return metric_definitions_; +} + +// .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; +inline void AlgorithmSpecification::clear_input_content_type() { + input_content_type_ = 0; +} +inline ::flyteidl::plugins::sagemaker::InputContentType_Value AlgorithmSpecification::input_content_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.AlgorithmSpecification.input_content_type) + return static_cast< ::flyteidl::plugins::sagemaker::InputContentType_Value >(input_content_type_); +} +inline void AlgorithmSpecification::set_input_content_type(::flyteidl::plugins::sagemaker::InputContentType_Value value) { + + input_content_type_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.AlgorithmSpecification.input_content_type) +} + +// ------------------------------------------------------------------- + +// TrainingJobResourceConfig + +// int64 instance_count = 1; +inline void TrainingJobResourceConfig::clear_instance_count() { + instance_count_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 TrainingJobResourceConfig::instance_count() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_count) + return instance_count_; +} +inline void TrainingJobResourceConfig::set_instance_count(::google::protobuf::int64 value) { + + instance_count_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_count) +} + +// string instance_type = 2; +inline void TrainingJobResourceConfig::clear_instance_type() { + instance_type_.ClearToEmptyNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline const ::std::string& TrainingJobResourceConfig::instance_type() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) + return instance_type_.GetNoArena(); +} +inline void TrainingJobResourceConfig::set_instance_type(const ::std::string& value) { + + instance_type_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), value); + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) +} +#if LANG_CXX11 +inline void TrainingJobResourceConfig::set_instance_type(::std::string&& value) { + + instance_type_.SetNoArena( + &::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::move(value)); + // @@protoc_insertion_point(field_set_rvalue:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) +} +#endif +inline void TrainingJobResourceConfig::set_instance_type(const char* value) { + GOOGLE_DCHECK(value != nullptr); + + instance_type_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), ::std::string(value)); + // @@protoc_insertion_point(field_set_char:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) +} +inline void TrainingJobResourceConfig::set_instance_type(const char* value, size_t size) { + + instance_type_.SetNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), + ::std::string(reinterpret_cast(value), size)); + // @@protoc_insertion_point(field_set_pointer:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) +} +inline ::std::string* TrainingJobResourceConfig::mutable_instance_type() { + + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) + return instance_type_.MutableNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline ::std::string* TrainingJobResourceConfig::release_instance_type() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) + + return instance_type_.ReleaseNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited()); +} +inline void TrainingJobResourceConfig::set_allocated_instance_type(::std::string* instance_type) { + if (instance_type != nullptr) { + + } else { + + } + instance_type_.SetAllocatedNoArena(&::google::protobuf::internal::GetEmptyStringAlreadyInited(), instance_type); + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type) +} + +// int64 volume_size_in_gb = 3; +inline void TrainingJobResourceConfig::clear_volume_size_in_gb() { + volume_size_in_gb_ = PROTOBUF_LONGLONG(0); +} +inline ::google::protobuf::int64 TrainingJobResourceConfig::volume_size_in_gb() const { + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.volume_size_in_gb) + return volume_size_in_gb_; +} +inline void TrainingJobResourceConfig::set_volume_size_in_gb(::google::protobuf::int64 value) { + + volume_size_in_gb_ = value; + // @@protoc_insertion_point(field_set:flyteidl.plugins.sagemaker.TrainingJobResourceConfig.volume_size_in_gb) +} + +// ------------------------------------------------------------------- + +// TrainingJob + +// .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; +inline bool TrainingJob::has_algorithm_specification() const { + return this != internal_default_instance() && algorithm_specification_ != nullptr; +} +inline void TrainingJob::clear_algorithm_specification() { + if (GetArenaNoVirtual() == nullptr && algorithm_specification_ != nullptr) { + delete algorithm_specification_; + } + algorithm_specification_ = nullptr; +} +inline const ::flyteidl::plugins::sagemaker::AlgorithmSpecification& TrainingJob::algorithm_specification() const { + const ::flyteidl::plugins::sagemaker::AlgorithmSpecification* p = algorithm_specification_; + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification) + return p != nullptr ? *p : *reinterpret_cast( + &::flyteidl::plugins::sagemaker::_AlgorithmSpecification_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::AlgorithmSpecification* TrainingJob::release_algorithm_specification() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification) + + ::flyteidl::plugins::sagemaker::AlgorithmSpecification* temp = algorithm_specification_; + algorithm_specification_ = nullptr; + return temp; +} +inline ::flyteidl::plugins::sagemaker::AlgorithmSpecification* TrainingJob::mutable_algorithm_specification() { + + if (algorithm_specification_ == nullptr) { + auto* p = CreateMaybeMessage<::flyteidl::plugins::sagemaker::AlgorithmSpecification>(GetArenaNoVirtual()); + algorithm_specification_ = p; + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification) + return algorithm_specification_; +} +inline void TrainingJob::set_allocated_algorithm_specification(::flyteidl::plugins::sagemaker::AlgorithmSpecification* algorithm_specification) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + if (message_arena == nullptr) { + delete algorithm_specification_; + } + if (algorithm_specification) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + algorithm_specification = ::google::protobuf::internal::GetOwnedMessage( + message_arena, algorithm_specification, submessage_arena); + } + + } else { + + } + algorithm_specification_ = algorithm_specification; + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification) +} + +// .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; +inline bool TrainingJob::has_training_job_resource_config() const { + return this != internal_default_instance() && training_job_resource_config_ != nullptr; +} +inline void TrainingJob::clear_training_job_resource_config() { + if (GetArenaNoVirtual() == nullptr && training_job_resource_config_ != nullptr) { + delete training_job_resource_config_; + } + training_job_resource_config_ = nullptr; +} +inline const ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig& TrainingJob::training_job_resource_config() const { + const ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* p = training_job_resource_config_; + // @@protoc_insertion_point(field_get:flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config) + return p != nullptr ? *p : *reinterpret_cast( + &::flyteidl::plugins::sagemaker::_TrainingJobResourceConfig_default_instance_); +} +inline ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* TrainingJob::release_training_job_resource_config() { + // @@protoc_insertion_point(field_release:flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config) + + ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* temp = training_job_resource_config_; + training_job_resource_config_ = nullptr; + return temp; +} +inline ::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* TrainingJob::mutable_training_job_resource_config() { + + if (training_job_resource_config_ == nullptr) { + auto* p = CreateMaybeMessage<::flyteidl::plugins::sagemaker::TrainingJobResourceConfig>(GetArenaNoVirtual()); + training_job_resource_config_ = p; + } + // @@protoc_insertion_point(field_mutable:flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config) + return training_job_resource_config_; +} +inline void TrainingJob::set_allocated_training_job_resource_config(::flyteidl::plugins::sagemaker::TrainingJobResourceConfig* training_job_resource_config) { + ::google::protobuf::Arena* message_arena = GetArenaNoVirtual(); + if (message_arena == nullptr) { + delete training_job_resource_config_; + } + if (training_job_resource_config) { + ::google::protobuf::Arena* submessage_arena = nullptr; + if (message_arena != submessage_arena) { + training_job_resource_config = ::google::protobuf::internal::GetOwnedMessage( + message_arena, training_job_resource_config, submessage_arena); + } + + } else { + + } + training_job_resource_config_ = training_job_resource_config; + // @@protoc_insertion_point(field_set_allocated:flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config) +} + +#ifdef __GNUC__ + #pragma GCC diagnostic pop +#endif // __GNUC__ +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + +// ------------------------------------------------------------------- + + +// @@protoc_insertion_point(namespace_scope) + +} // namespace sagemaker +} // namespace plugins +} // namespace flyteidl + +namespace google { +namespace protobuf { + +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::InputMode_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::InputMode_Value>() { + return ::flyteidl::plugins::sagemaker::InputMode_Value_descriptor(); +} +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::AlgorithmName_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::AlgorithmName_Value>() { + return ::flyteidl::plugins::sagemaker::AlgorithmName_Value_descriptor(); +} +template <> struct is_proto_enum< ::flyteidl::plugins::sagemaker::InputContentType_Value> : ::std::true_type {}; +template <> +inline const EnumDescriptor* GetEnumDescriptor< ::flyteidl::plugins::sagemaker::InputContentType_Value>() { + return ::flyteidl::plugins::sagemaker::InputContentType_Value_descriptor(); +} + +} // namespace protobuf +} // namespace google + +// @@protoc_insertion_point(global_scope) + +#include +#endif // PROTOBUF_INCLUDED_flyteidl_2fplugins_2fsagemaker_2ftraining_5fjob_2eproto diff --git a/gen/pb-go/flyteidl/core/compiler.pb.validate.go b/gen/pb-go/flyteidl/core/compiler.pb.validate.go index 7bb0ebff5..5fa2a39c3 100644 --- a/gen/pb-go/flyteidl/core/compiler.pb.validate.go +++ b/gen/pb-go/flyteidl/core/compiler.pb.validate.go @@ -44,9 +44,39 @@ func (m *ConnectionSet) Validate() error { return nil } - // no validation rules for Downstream + for key, val := range m.GetDownstream() { + _ = val - // no validation rules for Upstream + // no validation rules for Downstream[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ConnectionSetValidationError{ + field: fmt.Sprintf("Downstream[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } + + for key, val := range m.GetUpstream() { + _ = val + + // no validation rules for Upstream[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ConnectionSetValidationError{ + field: fmt.Sprintf("Upstream[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } return nil } diff --git a/gen/pb-go/flyteidl/core/interface.pb.validate.go b/gen/pb-go/flyteidl/core/interface.pb.validate.go index cc5a21288..6e51a5e88 100644 --- a/gen/pb-go/flyteidl/core/interface.pb.validate.go +++ b/gen/pb-go/flyteidl/core/interface.pb.validate.go @@ -120,7 +120,22 @@ func (m *VariableMap) Validate() error { return nil } - // no validation rules for Variables + for key, val := range m.GetVariables() { + _ = val + + // no validation rules for Variables[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return VariableMapValidationError{ + field: fmt.Sprintf("Variables[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } return nil } @@ -365,7 +380,22 @@ func (m *ParameterMap) Validate() error { return nil } - // no validation rules for Parameters + for key, val := range m.GetParameters() { + _ = val + + // no validation rules for Parameters[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ParameterMapValidationError{ + field: fmt.Sprintf("Parameters[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } return nil } diff --git a/gen/pb-go/flyteidl/core/literals.pb.validate.go b/gen/pb-go/flyteidl/core/literals.pb.validate.go index 212ed5d6e..8bf01c495 100644 --- a/gen/pb-go/flyteidl/core/literals.pb.validate.go +++ b/gen/pb-go/flyteidl/core/literals.pb.validate.go @@ -844,7 +844,22 @@ func (m *LiteralMap) Validate() error { return nil } - // no validation rules for Literals + for key, val := range m.GetLiterals() { + _ = val + + // no validation rules for Literals[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return LiteralMapValidationError{ + field: fmt.Sprintf("Literals[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } return nil } @@ -993,7 +1008,22 @@ func (m *BindingDataMap) Validate() error { return nil } - // no validation rules for Bindings + for key, val := range m.GetBindings() { + _ = val + + // no validation rules for Bindings[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return BindingDataMapValidationError{ + field: fmt.Sprintf("Bindings[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } return nil } diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.go b/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.go new file mode 100644 index 000000000..f990f056d --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.go @@ -0,0 +1,437 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +package plugins + +import ( + fmt "fmt" + proto "github.com/golang/protobuf/proto" + math "math" +) + +// Reference imports to suppress errors if they are not otherwise used. +var _ = proto.Marshal +var _ = fmt.Errorf +var _ = math.Inf + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the proto package it is being compiled against. +// A compilation error at this line likely means your copy of the +// proto package needs to be updated. +const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package + +type HyperparameterTuningObjectiveType_Value int32 + +const ( + HyperparameterTuningObjectiveType_MINIMIZE HyperparameterTuningObjectiveType_Value = 0 + HyperparameterTuningObjectiveType_MAXIMIZE HyperparameterTuningObjectiveType_Value = 1 +) + +var HyperparameterTuningObjectiveType_Value_name = map[int32]string{ + 0: "MINIMIZE", + 1: "MAXIMIZE", +} + +var HyperparameterTuningObjectiveType_Value_value = map[string]int32{ + "MINIMIZE": 0, + "MAXIMIZE": 1, +} + +func (x HyperparameterTuningObjectiveType_Value) String() string { + return proto.EnumName(HyperparameterTuningObjectiveType_Value_name, int32(x)) +} + +func (HyperparameterTuningObjectiveType_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{1, 0} +} + +type HyperparameterTuningStrategy_Value int32 + +const ( + HyperparameterTuningStrategy_BAYESIAN HyperparameterTuningStrategy_Value = 0 + HyperparameterTuningStrategy_RANDOM HyperparameterTuningStrategy_Value = 1 +) + +var HyperparameterTuningStrategy_Value_name = map[int32]string{ + 0: "BAYESIAN", + 1: "RANDOM", +} + +var HyperparameterTuningStrategy_Value_value = map[string]int32{ + "BAYESIAN": 0, + "RANDOM": 1, +} + +func (x HyperparameterTuningStrategy_Value) String() string { + return proto.EnumName(HyperparameterTuningStrategy_Value_name, int32(x)) +} + +func (HyperparameterTuningStrategy_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{3, 0} +} + +type TrainingJobEarlyStoppingType_Value int32 + +const ( + TrainingJobEarlyStoppingType_OFF TrainingJobEarlyStoppingType_Value = 0 + TrainingJobEarlyStoppingType_AUTO TrainingJobEarlyStoppingType_Value = 1 +) + +var TrainingJobEarlyStoppingType_Value_name = map[int32]string{ + 0: "OFF", + 1: "AUTO", +} + +var TrainingJobEarlyStoppingType_Value_value = map[string]int32{ + "OFF": 0, + "AUTO": 1, +} + +func (x TrainingJobEarlyStoppingType_Value) String() string { + return proto.EnumName(TrainingJobEarlyStoppingType_Value_name, int32(x)) +} + +func (TrainingJobEarlyStoppingType_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{4, 0} +} + +// A pass-through for SageMaker's hyperparameter tuning job +type HyperparameterTuningJob struct { + // The underlying training job that the hyperparameter tuning job will launch during the process + TrainingJob *TrainingJob `protobuf:"bytes,1,opt,name=training_job,json=trainingJob,proto3" json:"training_job,omitempty"` + // The maximum number of training jobs that an hpo job can launch. For resource limit purpose. + // https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + MaxNumberOfTrainingJobs int64 `protobuf:"varint,2,opt,name=max_number_of_training_jobs,json=maxNumberOfTrainingJobs,proto3" json:"max_number_of_training_jobs,omitempty"` + // The maximum number of concurrent training job that an hpo job can launch + // https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + MaxParallelTrainingJobs int64 `protobuf:"varint,3,opt,name=max_parallel_training_jobs,json=maxParallelTrainingJobs,proto3" json:"max_parallel_training_jobs,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterTuningJob) Reset() { *m = HyperparameterTuningJob{} } +func (m *HyperparameterTuningJob) String() string { return proto.CompactTextString(m) } +func (*HyperparameterTuningJob) ProtoMessage() {} +func (*HyperparameterTuningJob) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{0} +} + +func (m *HyperparameterTuningJob) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterTuningJob.Unmarshal(m, b) +} +func (m *HyperparameterTuningJob) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterTuningJob.Marshal(b, m, deterministic) +} +func (m *HyperparameterTuningJob) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterTuningJob.Merge(m, src) +} +func (m *HyperparameterTuningJob) XXX_Size() int { + return xxx_messageInfo_HyperparameterTuningJob.Size(m) +} +func (m *HyperparameterTuningJob) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterTuningJob.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterTuningJob proto.InternalMessageInfo + +func (m *HyperparameterTuningJob) GetTrainingJob() *TrainingJob { + if m != nil { + return m.TrainingJob + } + return nil +} + +func (m *HyperparameterTuningJob) GetMaxNumberOfTrainingJobs() int64 { + if m != nil { + return m.MaxNumberOfTrainingJobs + } + return 0 +} + +func (m *HyperparameterTuningJob) GetMaxParallelTrainingJobs() int64 { + if m != nil { + return m.MaxParallelTrainingJobs + } + return 0 +} + +// HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job +// with respect to the specified metric. +type HyperparameterTuningObjectiveType struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterTuningObjectiveType) Reset() { *m = HyperparameterTuningObjectiveType{} } +func (m *HyperparameterTuningObjectiveType) String() string { return proto.CompactTextString(m) } +func (*HyperparameterTuningObjectiveType) ProtoMessage() {} +func (*HyperparameterTuningObjectiveType) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{1} +} + +func (m *HyperparameterTuningObjectiveType) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterTuningObjectiveType.Unmarshal(m, b) +} +func (m *HyperparameterTuningObjectiveType) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterTuningObjectiveType.Marshal(b, m, deterministic) +} +func (m *HyperparameterTuningObjectiveType) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterTuningObjectiveType.Merge(m, src) +} +func (m *HyperparameterTuningObjectiveType) XXX_Size() int { + return xxx_messageInfo_HyperparameterTuningObjectiveType.Size(m) +} +func (m *HyperparameterTuningObjectiveType) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterTuningObjectiveType.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterTuningObjectiveType proto.InternalMessageInfo + +// The target metric and the objective of the hyperparameter tuning. +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html +type HyperparameterTuningObjective struct { + // HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job + // with respect to the specified metric. + ObjectiveType HyperparameterTuningObjectiveType_Value `protobuf:"varint,1,opt,name=objective_type,json=objectiveType,proto3,enum=flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType_Value" json:"objective_type,omitempty"` + // The target metric name, which is the user-defined name of the metric specified in the + // training job's algorithm specification + MetricName string `protobuf:"bytes,2,opt,name=metric_name,json=metricName,proto3" json:"metric_name,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterTuningObjective) Reset() { *m = HyperparameterTuningObjective{} } +func (m *HyperparameterTuningObjective) String() string { return proto.CompactTextString(m) } +func (*HyperparameterTuningObjective) ProtoMessage() {} +func (*HyperparameterTuningObjective) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{2} +} + +func (m *HyperparameterTuningObjective) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterTuningObjective.Unmarshal(m, b) +} +func (m *HyperparameterTuningObjective) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterTuningObjective.Marshal(b, m, deterministic) +} +func (m *HyperparameterTuningObjective) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterTuningObjective.Merge(m, src) +} +func (m *HyperparameterTuningObjective) XXX_Size() int { + return xxx_messageInfo_HyperparameterTuningObjective.Size(m) +} +func (m *HyperparameterTuningObjective) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterTuningObjective.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterTuningObjective proto.InternalMessageInfo + +func (m *HyperparameterTuningObjective) GetObjectiveType() HyperparameterTuningObjectiveType_Value { + if m != nil { + return m.ObjectiveType + } + return HyperparameterTuningObjectiveType_MINIMIZE +} + +func (m *HyperparameterTuningObjective) GetMetricName() string { + if m != nil { + return m.MetricName + } + return "" +} + +// Setting the strategy used when searching in the hyperparameter space +// Refer this doc for more details: +// https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ +type HyperparameterTuningStrategy struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterTuningStrategy) Reset() { *m = HyperparameterTuningStrategy{} } +func (m *HyperparameterTuningStrategy) String() string { return proto.CompactTextString(m) } +func (*HyperparameterTuningStrategy) ProtoMessage() {} +func (*HyperparameterTuningStrategy) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{3} +} + +func (m *HyperparameterTuningStrategy) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterTuningStrategy.Unmarshal(m, b) +} +func (m *HyperparameterTuningStrategy) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterTuningStrategy.Marshal(b, m, deterministic) +} +func (m *HyperparameterTuningStrategy) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterTuningStrategy.Merge(m, src) +} +func (m *HyperparameterTuningStrategy) XXX_Size() int { + return xxx_messageInfo_HyperparameterTuningStrategy.Size(m) +} +func (m *HyperparameterTuningStrategy) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterTuningStrategy.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterTuningStrategy proto.InternalMessageInfo + +// When the training jobs launched by the hyperparameter tuning job are not improving significantly, +// a hyperparameter tuning job can be stopping early. +// Note that there's only a subset of built-in algorithms that supports early stopping. +// see: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-early-stopping.html +type TrainingJobEarlyStoppingType struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *TrainingJobEarlyStoppingType) Reset() { *m = TrainingJobEarlyStoppingType{} } +func (m *TrainingJobEarlyStoppingType) String() string { return proto.CompactTextString(m) } +func (*TrainingJobEarlyStoppingType) ProtoMessage() {} +func (*TrainingJobEarlyStoppingType) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{4} +} + +func (m *TrainingJobEarlyStoppingType) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_TrainingJobEarlyStoppingType.Unmarshal(m, b) +} +func (m *TrainingJobEarlyStoppingType) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_TrainingJobEarlyStoppingType.Marshal(b, m, deterministic) +} +func (m *TrainingJobEarlyStoppingType) XXX_Merge(src proto.Message) { + xxx_messageInfo_TrainingJobEarlyStoppingType.Merge(m, src) +} +func (m *TrainingJobEarlyStoppingType) XXX_Size() int { + return xxx_messageInfo_TrainingJobEarlyStoppingType.Size(m) +} +func (m *TrainingJobEarlyStoppingType) XXX_DiscardUnknown() { + xxx_messageInfo_TrainingJobEarlyStoppingType.DiscardUnknown(m) +} + +var xxx_messageInfo_TrainingJobEarlyStoppingType proto.InternalMessageInfo + +// The specification of the hyperparameter tuning process +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config +type HyperparameterTuningJobConfig struct { + // ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range + HyperparameterRanges *ParameterRanges `protobuf:"bytes,1,opt,name=hyperparameter_ranges,json=hyperparameterRanges,proto3" json:"hyperparameter_ranges,omitempty"` + // Setting the strategy used when searching in the hyperparameter space + TuningStrategy HyperparameterTuningStrategy_Value `protobuf:"varint,2,opt,name=tuning_strategy,json=tuningStrategy,proto3,enum=flyteidl.plugins.sagemaker.HyperparameterTuningStrategy_Value" json:"tuning_strategy,omitempty"` + // The target metric and the objective of the hyperparameter tuning. + TuningObjective *HyperparameterTuningObjective `protobuf:"bytes,3,opt,name=tuning_objective,json=tuningObjective,proto3" json:"tuning_objective,omitempty"` + // When the training jobs launched by the hyperparameter tuning job are not improving significantly, + // a hyperparameter tuning job can be stopping early. + TrainingJobEarlyStoppingType TrainingJobEarlyStoppingType_Value `protobuf:"varint,4,opt,name=training_job_early_stopping_type,json=trainingJobEarlyStoppingType,proto3,enum=flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType_Value" json:"training_job_early_stopping_type,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterTuningJobConfig) Reset() { *m = HyperparameterTuningJobConfig{} } +func (m *HyperparameterTuningJobConfig) String() string { return proto.CompactTextString(m) } +func (*HyperparameterTuningJobConfig) ProtoMessage() {} +func (*HyperparameterTuningJobConfig) Descriptor() ([]byte, []int) { + return fileDescriptor_84374f4d1322c4ba, []int{5} +} + +func (m *HyperparameterTuningJobConfig) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterTuningJobConfig.Unmarshal(m, b) +} +func (m *HyperparameterTuningJobConfig) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterTuningJobConfig.Marshal(b, m, deterministic) +} +func (m *HyperparameterTuningJobConfig) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterTuningJobConfig.Merge(m, src) +} +func (m *HyperparameterTuningJobConfig) XXX_Size() int { + return xxx_messageInfo_HyperparameterTuningJobConfig.Size(m) +} +func (m *HyperparameterTuningJobConfig) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterTuningJobConfig.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterTuningJobConfig proto.InternalMessageInfo + +func (m *HyperparameterTuningJobConfig) GetHyperparameterRanges() *ParameterRanges { + if m != nil { + return m.HyperparameterRanges + } + return nil +} + +func (m *HyperparameterTuningJobConfig) GetTuningStrategy() HyperparameterTuningStrategy_Value { + if m != nil { + return m.TuningStrategy + } + return HyperparameterTuningStrategy_BAYESIAN +} + +func (m *HyperparameterTuningJobConfig) GetTuningObjective() *HyperparameterTuningObjective { + if m != nil { + return m.TuningObjective + } + return nil +} + +func (m *HyperparameterTuningJobConfig) GetTrainingJobEarlyStoppingType() TrainingJobEarlyStoppingType_Value { + if m != nil { + return m.TrainingJobEarlyStoppingType + } + return TrainingJobEarlyStoppingType_OFF +} + +func init() { + proto.RegisterEnum("flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType_Value", HyperparameterTuningObjectiveType_Value_name, HyperparameterTuningObjectiveType_Value_value) + proto.RegisterEnum("flyteidl.plugins.sagemaker.HyperparameterTuningStrategy_Value", HyperparameterTuningStrategy_Value_name, HyperparameterTuningStrategy_Value_value) + proto.RegisterEnum("flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType_Value", TrainingJobEarlyStoppingType_Value_name, TrainingJobEarlyStoppingType_Value_value) + proto.RegisterType((*HyperparameterTuningJob)(nil), "flyteidl.plugins.sagemaker.HyperparameterTuningJob") + proto.RegisterType((*HyperparameterTuningObjectiveType)(nil), "flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType") + proto.RegisterType((*HyperparameterTuningObjective)(nil), "flyteidl.plugins.sagemaker.HyperparameterTuningObjective") + proto.RegisterType((*HyperparameterTuningStrategy)(nil), "flyteidl.plugins.sagemaker.HyperparameterTuningStrategy") + proto.RegisterType((*TrainingJobEarlyStoppingType)(nil), "flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType") + proto.RegisterType((*HyperparameterTuningJobConfig)(nil), "flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig") +} + +func init() { + proto.RegisterFile("flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto", fileDescriptor_84374f4d1322c4ba) +} + +var fileDescriptor_84374f4d1322c4ba = []byte{ + // 554 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x9c, 0x54, 0xc1, 0x6e, 0xd3, 0x40, + 0x10, 0xad, 0x49, 0x29, 0x65, 0x52, 0x82, 0xb5, 0x02, 0xb5, 0x0a, 0x41, 0xa4, 0xe6, 0x40, 0x25, + 0x54, 0x5b, 0xa4, 0xe2, 0x40, 0x41, 0x48, 0x6e, 0x49, 0xd5, 0x44, 0x4a, 0x52, 0x39, 0x01, 0x41, + 0x2f, 0x66, 0x9d, 0x6e, 0xb6, 0x0e, 0xb6, 0xd7, 0x5a, 0x6f, 0x50, 0xfd, 0x03, 0x7c, 0x0d, 0x3f, + 0xc4, 0x8d, 0x4f, 0x41, 0x59, 0x6f, 0x12, 0x27, 0x0a, 0xae, 0xe0, 0x38, 0xa3, 0x79, 0xcf, 0x6f, + 0xde, 0x1b, 0x2f, 0x1c, 0x8f, 0x82, 0x54, 0x10, 0xff, 0x2a, 0xb0, 0xe2, 0x60, 0x42, 0xfd, 0x28, + 0xb1, 0x12, 0x4c, 0x49, 0x88, 0xbf, 0x11, 0x6e, 0x5d, 0xa7, 0x31, 0xe1, 0x31, 0xe6, 0x38, 0x24, + 0x82, 0x70, 0x57, 0x4c, 0x22, 0x3f, 0xa2, 0xee, 0x98, 0x79, 0x66, 0xcc, 0x99, 0x60, 0xa8, 0x3a, + 0xc3, 0x9a, 0x0a, 0x6b, 0xce, 0xb1, 0xd5, 0x57, 0x05, 0xbc, 0x0b, 0x4a, 0x8e, 0x23, 0x4a, 0x92, + 0x8c, 0xae, 0x7a, 0x58, 0x00, 0x11, 0x1c, 0xfb, 0xcb, 0x5f, 0x37, 0x7e, 0x6b, 0xb0, 0x7b, 0xbe, + 0xa4, 0x70, 0x20, 0x05, 0xb6, 0x99, 0x87, 0xda, 0xb0, 0x93, 0x47, 0xec, 0x69, 0x75, 0xed, 0xa0, + 0xdc, 0x78, 0x61, 0xfe, 0x5d, 0xb0, 0x39, 0x50, 0xf3, 0x6d, 0xe6, 0x39, 0x65, 0xb1, 0x28, 0xd0, + 0x3b, 0x78, 0x12, 0xe2, 0x1b, 0x37, 0x9a, 0x84, 0x1e, 0xe1, 0x2e, 0x1b, 0xb9, 0x79, 0xe6, 0x64, + 0xef, 0x4e, 0x5d, 0x3b, 0x28, 0x39, 0xbb, 0x21, 0xbe, 0xe9, 0xca, 0x89, 0xde, 0x28, 0xc7, 0x94, + 0xa0, 0xb7, 0x50, 0x9d, 0xa2, 0xa7, 0x1a, 0x83, 0x80, 0x04, 0x2b, 0xe0, 0xd2, 0x1c, 0x7c, 0xa1, + 0x06, 0xf2, 0x60, 0xe3, 0x1c, 0xf6, 0xd7, 0x6d, 0xd8, 0xf3, 0xc6, 0x64, 0x28, 0xfc, 0xef, 0x64, + 0x90, 0xc6, 0xc4, 0x78, 0x0e, 0x77, 0x3f, 0xe1, 0x60, 0x42, 0xd0, 0x0e, 0x6c, 0x77, 0x5a, 0xdd, + 0x56, 0xa7, 0x75, 0xd9, 0xd4, 0x37, 0x64, 0x65, 0x7f, 0xce, 0x2a, 0xcd, 0xf8, 0xa9, 0xc1, 0xd3, + 0x42, 0x2a, 0x34, 0x86, 0x0a, 0x9b, 0x15, 0xae, 0x48, 0x63, 0x22, 0x4d, 0xab, 0x34, 0x4e, 0x8b, + 0x4c, 0xbb, 0x55, 0x9d, 0x29, 0xa5, 0x39, 0x0f, 0x58, 0xbe, 0x89, 0x9e, 0x41, 0x39, 0x24, 0x82, + 0xfb, 0x43, 0x37, 0xc2, 0x21, 0x91, 0x16, 0xde, 0x77, 0x20, 0x6b, 0x75, 0x71, 0x48, 0x0c, 0x1b, + 0x6a, 0xeb, 0xa8, 0xfb, 0x82, 0x63, 0x41, 0x68, 0x6a, 0xec, 0xe7, 0x76, 0x3e, 0xb1, 0xbf, 0x34, + 0xfb, 0x2d, 0xbb, 0xab, 0x6f, 0x20, 0x80, 0x2d, 0xc7, 0xee, 0x7e, 0xe8, 0x75, 0x74, 0xcd, 0x38, + 0x86, 0x5a, 0xce, 0xcb, 0x26, 0xe6, 0x41, 0xda, 0x17, 0x2c, 0x8e, 0xfd, 0x88, 0x4a, 0xdb, 0xaa, + 0x33, 0x8a, 0x7b, 0x50, 0xea, 0x9d, 0x9d, 0xe9, 0x1b, 0x68, 0x1b, 0x36, 0xed, 0x8f, 0x83, 0x9e, + 0xae, 0x19, 0xbf, 0x4a, 0xeb, 0xdd, 0x6a, 0x33, 0xef, 0x94, 0x45, 0x23, 0x9f, 0xa2, 0xaf, 0xf0, + 0x78, 0xe5, 0xef, 0xc8, 0x4e, 0x59, 0x5d, 0xda, 0xcb, 0x22, 0xd3, 0x2e, 0x66, 0x18, 0x47, 0x42, + 0x9c, 0x47, 0xcb, 0x4c, 0x59, 0x17, 0x51, 0x78, 0xa8, 0x7e, 0xb8, 0x44, 0x6d, 0x2d, 0x7d, 0xaa, + 0x34, 0xde, 0xff, 0x6b, 0x20, 0x33, 0xd7, 0x54, 0x16, 0x15, 0xb1, 0xd4, 0x45, 0x57, 0xa0, 0xab, + 0x0f, 0xcd, 0x43, 0x92, 0x77, 0x59, 0x6e, 0xbc, 0xf9, 0xef, 0xe8, 0x1d, 0xa5, 0x7d, 0x71, 0x5e, + 0x3f, 0x34, 0xa8, 0xe7, 0x6f, 0xdf, 0x25, 0xd3, 0x44, 0xdc, 0x44, 0x45, 0x92, 0x5d, 0xdc, 0xe6, + 0xed, 0x0b, 0x16, 0x65, 0xaa, 0x16, 0xac, 0x89, 0x82, 0x99, 0x93, 0xd7, 0x97, 0x47, 0xd4, 0x17, + 0xd7, 0x13, 0xcf, 0x1c, 0xb2, 0xd0, 0x0a, 0xd2, 0x91, 0xb0, 0xe6, 0xef, 0x0e, 0x25, 0x91, 0x15, + 0x7b, 0x87, 0x94, 0x59, 0xab, 0x4f, 0x91, 0xb7, 0x25, 0x1f, 0x9d, 0xa3, 0x3f, 0x01, 0x00, 0x00, + 0xff, 0xff, 0x80, 0x43, 0xe0, 0x75, 0x30, 0x05, 0x00, 0x00, +} diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.validate.go b/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.validate.go new file mode 100644 index 000000000..08e5ef6b0 --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.pb.validate.go @@ -0,0 +1,486 @@ +// Code generated by protoc-gen-validate. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +package plugins + +import ( + "bytes" + "errors" + "fmt" + "net" + "net/mail" + "net/url" + "regexp" + "strings" + "time" + "unicode/utf8" + + "github.com/golang/protobuf/ptypes" +) + +// ensure the imports are used +var ( + _ = bytes.MinRead + _ = errors.New("") + _ = fmt.Print + _ = utf8.UTFMax + _ = (*regexp.Regexp)(nil) + _ = (*strings.Reader)(nil) + _ = net.IPv4len + _ = time.Duration(0) + _ = (*url.URL)(nil) + _ = (*mail.Address)(nil) + _ = ptypes.DynamicAny{} +) + +// define the regex for a UUID once up-front +var _hyperparameter_tuning_job_uuidPattern = regexp.MustCompile("^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$") + +// Validate checks the field values on HyperparameterTuningJob with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *HyperparameterTuningJob) Validate() error { + if m == nil { + return nil + } + + if v, ok := interface{}(m.GetTrainingJob()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return HyperparameterTuningJobValidationError{ + field: "TrainingJob", + reason: "embedded message failed validation", + cause: err, + } + } + } + + // no validation rules for MaxNumberOfTrainingJobs + + // no validation rules for MaxParallelTrainingJobs + + return nil +} + +// HyperparameterTuningJobValidationError is the validation error returned by +// HyperparameterTuningJob.Validate if the designated constraints aren't met. +type HyperparameterTuningJobValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterTuningJobValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterTuningJobValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterTuningJobValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterTuningJobValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterTuningJobValidationError) ErrorName() string { + return "HyperparameterTuningJobValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterTuningJobValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterTuningJob.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterTuningJobValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterTuningJobValidationError{} + +// Validate checks the field values on HyperparameterTuningObjectiveType with +// the rules defined in the proto definition for this message. If any rules +// are violated, an error is returned. +func (m *HyperparameterTuningObjectiveType) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// HyperparameterTuningObjectiveTypeValidationError is the validation error +// returned by HyperparameterTuningObjectiveType.Validate if the designated +// constraints aren't met. +type HyperparameterTuningObjectiveTypeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterTuningObjectiveTypeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterTuningObjectiveTypeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterTuningObjectiveTypeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterTuningObjectiveTypeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterTuningObjectiveTypeValidationError) ErrorName() string { + return "HyperparameterTuningObjectiveTypeValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterTuningObjectiveTypeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterTuningObjectiveType.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterTuningObjectiveTypeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterTuningObjectiveTypeValidationError{} + +// Validate checks the field values on HyperparameterTuningObjective with the +// rules defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *HyperparameterTuningObjective) Validate() error { + if m == nil { + return nil + } + + // no validation rules for ObjectiveType + + // no validation rules for MetricName + + return nil +} + +// HyperparameterTuningObjectiveValidationError is the validation error +// returned by HyperparameterTuningObjective.Validate if the designated +// constraints aren't met. +type HyperparameterTuningObjectiveValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterTuningObjectiveValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterTuningObjectiveValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterTuningObjectiveValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterTuningObjectiveValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterTuningObjectiveValidationError) ErrorName() string { + return "HyperparameterTuningObjectiveValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterTuningObjectiveValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterTuningObjective.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterTuningObjectiveValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterTuningObjectiveValidationError{} + +// Validate checks the field values on HyperparameterTuningStrategy with the +// rules defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *HyperparameterTuningStrategy) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// HyperparameterTuningStrategyValidationError is the validation error returned +// by HyperparameterTuningStrategy.Validate if the designated constraints +// aren't met. +type HyperparameterTuningStrategyValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterTuningStrategyValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterTuningStrategyValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterTuningStrategyValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterTuningStrategyValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterTuningStrategyValidationError) ErrorName() string { + return "HyperparameterTuningStrategyValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterTuningStrategyValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterTuningStrategy.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterTuningStrategyValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterTuningStrategyValidationError{} + +// Validate checks the field values on TrainingJobEarlyStoppingType with the +// rules defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *TrainingJobEarlyStoppingType) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// TrainingJobEarlyStoppingTypeValidationError is the validation error returned +// by TrainingJobEarlyStoppingType.Validate if the designated constraints +// aren't met. +type TrainingJobEarlyStoppingTypeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e TrainingJobEarlyStoppingTypeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e TrainingJobEarlyStoppingTypeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e TrainingJobEarlyStoppingTypeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e TrainingJobEarlyStoppingTypeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e TrainingJobEarlyStoppingTypeValidationError) ErrorName() string { + return "TrainingJobEarlyStoppingTypeValidationError" +} + +// Error satisfies the builtin error interface +func (e TrainingJobEarlyStoppingTypeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sTrainingJobEarlyStoppingType.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = TrainingJobEarlyStoppingTypeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = TrainingJobEarlyStoppingTypeValidationError{} + +// Validate checks the field values on HyperparameterTuningJobConfig with the +// rules defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *HyperparameterTuningJobConfig) Validate() error { + if m == nil { + return nil + } + + if v, ok := interface{}(m.GetHyperparameterRanges()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return HyperparameterTuningJobConfigValidationError{ + field: "HyperparameterRanges", + reason: "embedded message failed validation", + cause: err, + } + } + } + + // no validation rules for TuningStrategy + + if v, ok := interface{}(m.GetTuningObjective()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return HyperparameterTuningJobConfigValidationError{ + field: "TuningObjective", + reason: "embedded message failed validation", + cause: err, + } + } + } + + // no validation rules for TrainingJobEarlyStoppingType + + return nil +} + +// HyperparameterTuningJobConfigValidationError is the validation error +// returned by HyperparameterTuningJobConfig.Validate if the designated +// constraints aren't met. +type HyperparameterTuningJobConfigValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterTuningJobConfigValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterTuningJobConfigValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterTuningJobConfigValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterTuningJobConfigValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterTuningJobConfigValidationError) ErrorName() string { + return "HyperparameterTuningJobConfigValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterTuningJobConfigValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterTuningJobConfig.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterTuningJobConfigValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterTuningJobConfigValidationError{} diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.go b/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.go new file mode 100644 index 000000000..2c61347b0 --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.go @@ -0,0 +1,435 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +package plugins + +import ( + fmt "fmt" + proto "github.com/golang/protobuf/proto" + math "math" +) + +// Reference imports to suppress errors if they are not otherwise used. +var _ = proto.Marshal +var _ = fmt.Errorf +var _ = math.Inf + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the proto package it is being compiled against. +// A compilation error at this line likely means your copy of the +// proto package needs to be updated. +const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package + +type HyperparameterScalingType_Value int32 + +const ( + HyperparameterScalingType_AUTO HyperparameterScalingType_Value = 0 + HyperparameterScalingType_LINEAR HyperparameterScalingType_Value = 1 + HyperparameterScalingType_LOGARITHMIC HyperparameterScalingType_Value = 2 + HyperparameterScalingType_REVERSELOGARITHMIC HyperparameterScalingType_Value = 3 +) + +var HyperparameterScalingType_Value_name = map[int32]string{ + 0: "AUTO", + 1: "LINEAR", + 2: "LOGARITHMIC", + 3: "REVERSELOGARITHMIC", +} + +var HyperparameterScalingType_Value_value = map[string]int32{ + "AUTO": 0, + "LINEAR": 1, + "LOGARITHMIC": 2, + "REVERSELOGARITHMIC": 3, +} + +func (x HyperparameterScalingType_Value) String() string { + return proto.EnumName(HyperparameterScalingType_Value_name, int32(x)) +} + +func (HyperparameterScalingType_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{0, 0} +} + +// HyperparameterScalingType defines the way to increase or decrease the value of the hyperparameter +// For details, refer to: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +// See examples of these scaling type, refer to: https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ +type HyperparameterScalingType struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *HyperparameterScalingType) Reset() { *m = HyperparameterScalingType{} } +func (m *HyperparameterScalingType) String() string { return proto.CompactTextString(m) } +func (*HyperparameterScalingType) ProtoMessage() {} +func (*HyperparameterScalingType) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{0} +} + +func (m *HyperparameterScalingType) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_HyperparameterScalingType.Unmarshal(m, b) +} +func (m *HyperparameterScalingType) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_HyperparameterScalingType.Marshal(b, m, deterministic) +} +func (m *HyperparameterScalingType) XXX_Merge(src proto.Message) { + xxx_messageInfo_HyperparameterScalingType.Merge(m, src) +} +func (m *HyperparameterScalingType) XXX_Size() int { + return xxx_messageInfo_HyperparameterScalingType.Size(m) +} +func (m *HyperparameterScalingType) XXX_DiscardUnknown() { + xxx_messageInfo_HyperparameterScalingType.DiscardUnknown(m) +} + +var xxx_messageInfo_HyperparameterScalingType proto.InternalMessageInfo + +// ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +// users to specify the search space of a floating-point hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +type ContinuousParameterRange struct { + MaxValue float64 `protobuf:"fixed64,1,opt,name=max_value,json=maxValue,proto3" json:"max_value,omitempty"` + MinValue float64 `protobuf:"fixed64,2,opt,name=min_value,json=minValue,proto3" json:"min_value,omitempty"` + ScalingType HyperparameterScalingType_Value `protobuf:"varint,3,opt,name=scaling_type,json=scalingType,proto3,enum=flyteidl.plugins.sagemaker.HyperparameterScalingType_Value" json:"scaling_type,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *ContinuousParameterRange) Reset() { *m = ContinuousParameterRange{} } +func (m *ContinuousParameterRange) String() string { return proto.CompactTextString(m) } +func (*ContinuousParameterRange) ProtoMessage() {} +func (*ContinuousParameterRange) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{1} +} + +func (m *ContinuousParameterRange) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_ContinuousParameterRange.Unmarshal(m, b) +} +func (m *ContinuousParameterRange) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_ContinuousParameterRange.Marshal(b, m, deterministic) +} +func (m *ContinuousParameterRange) XXX_Merge(src proto.Message) { + xxx_messageInfo_ContinuousParameterRange.Merge(m, src) +} +func (m *ContinuousParameterRange) XXX_Size() int { + return xxx_messageInfo_ContinuousParameterRange.Size(m) +} +func (m *ContinuousParameterRange) XXX_DiscardUnknown() { + xxx_messageInfo_ContinuousParameterRange.DiscardUnknown(m) +} + +var xxx_messageInfo_ContinuousParameterRange proto.InternalMessageInfo + +func (m *ContinuousParameterRange) GetMaxValue() float64 { + if m != nil { + return m.MaxValue + } + return 0 +} + +func (m *ContinuousParameterRange) GetMinValue() float64 { + if m != nil { + return m.MinValue + } + return 0 +} + +func (m *ContinuousParameterRange) GetScalingType() HyperparameterScalingType_Value { + if m != nil { + return m.ScalingType + } + return HyperparameterScalingType_AUTO +} + +// IntegerParameterRange refers to a discrete range of hyperparameter values, allowing +// users to specify the search space of an integer hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +type IntegerParameterRange struct { + MaxValue int64 `protobuf:"varint,1,opt,name=max_value,json=maxValue,proto3" json:"max_value,omitempty"` + MinValue int64 `protobuf:"varint,2,opt,name=min_value,json=minValue,proto3" json:"min_value,omitempty"` + ScalingType HyperparameterScalingType_Value `protobuf:"varint,3,opt,name=scaling_type,json=scalingType,proto3,enum=flyteidl.plugins.sagemaker.HyperparameterScalingType_Value" json:"scaling_type,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *IntegerParameterRange) Reset() { *m = IntegerParameterRange{} } +func (m *IntegerParameterRange) String() string { return proto.CompactTextString(m) } +func (*IntegerParameterRange) ProtoMessage() {} +func (*IntegerParameterRange) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{2} +} + +func (m *IntegerParameterRange) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_IntegerParameterRange.Unmarshal(m, b) +} +func (m *IntegerParameterRange) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_IntegerParameterRange.Marshal(b, m, deterministic) +} +func (m *IntegerParameterRange) XXX_Merge(src proto.Message) { + xxx_messageInfo_IntegerParameterRange.Merge(m, src) +} +func (m *IntegerParameterRange) XXX_Size() int { + return xxx_messageInfo_IntegerParameterRange.Size(m) +} +func (m *IntegerParameterRange) XXX_DiscardUnknown() { + xxx_messageInfo_IntegerParameterRange.DiscardUnknown(m) +} + +var xxx_messageInfo_IntegerParameterRange proto.InternalMessageInfo + +func (m *IntegerParameterRange) GetMaxValue() int64 { + if m != nil { + return m.MaxValue + } + return 0 +} + +func (m *IntegerParameterRange) GetMinValue() int64 { + if m != nil { + return m.MinValue + } + return 0 +} + +func (m *IntegerParameterRange) GetScalingType() HyperparameterScalingType_Value { + if m != nil { + return m.ScalingType + } + return HyperparameterScalingType_AUTO +} + +// ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +// users to specify the search space of a floating-point hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +type CategoricalParameterRange struct { + Values []string `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *CategoricalParameterRange) Reset() { *m = CategoricalParameterRange{} } +func (m *CategoricalParameterRange) String() string { return proto.CompactTextString(m) } +func (*CategoricalParameterRange) ProtoMessage() {} +func (*CategoricalParameterRange) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{3} +} + +func (m *CategoricalParameterRange) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_CategoricalParameterRange.Unmarshal(m, b) +} +func (m *CategoricalParameterRange) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_CategoricalParameterRange.Marshal(b, m, deterministic) +} +func (m *CategoricalParameterRange) XXX_Merge(src proto.Message) { + xxx_messageInfo_CategoricalParameterRange.Merge(m, src) +} +func (m *CategoricalParameterRange) XXX_Size() int { + return xxx_messageInfo_CategoricalParameterRange.Size(m) +} +func (m *CategoricalParameterRange) XXX_DiscardUnknown() { + xxx_messageInfo_CategoricalParameterRange.DiscardUnknown(m) +} + +var xxx_messageInfo_CategoricalParameterRange proto.InternalMessageInfo + +func (m *CategoricalParameterRange) GetValues() []string { + if m != nil { + return m.Values + } + return nil +} + +// ParameterRangeOneOf describes a single ParameterRange, which is a one-of structure that can be one of +// the three possible types: ContinuousParameterRange, IntegerParameterRange, and CategoricalParameterRange. +// This one-of structure in Flyte enables specifying a Parameter in a type-safe manner +// See: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +type ParameterRangeOneOf struct { + // Types that are valid to be assigned to ParameterRangeType: + // *ParameterRangeOneOf_ContinuousParameterRange + // *ParameterRangeOneOf_IntegerParameterRange + // *ParameterRangeOneOf_CategoricalParameterRange + ParameterRangeType isParameterRangeOneOf_ParameterRangeType `protobuf_oneof:"parameter_range_type"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *ParameterRangeOneOf) Reset() { *m = ParameterRangeOneOf{} } +func (m *ParameterRangeOneOf) String() string { return proto.CompactTextString(m) } +func (*ParameterRangeOneOf) ProtoMessage() {} +func (*ParameterRangeOneOf) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{4} +} + +func (m *ParameterRangeOneOf) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_ParameterRangeOneOf.Unmarshal(m, b) +} +func (m *ParameterRangeOneOf) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_ParameterRangeOneOf.Marshal(b, m, deterministic) +} +func (m *ParameterRangeOneOf) XXX_Merge(src proto.Message) { + xxx_messageInfo_ParameterRangeOneOf.Merge(m, src) +} +func (m *ParameterRangeOneOf) XXX_Size() int { + return xxx_messageInfo_ParameterRangeOneOf.Size(m) +} +func (m *ParameterRangeOneOf) XXX_DiscardUnknown() { + xxx_messageInfo_ParameterRangeOneOf.DiscardUnknown(m) +} + +var xxx_messageInfo_ParameterRangeOneOf proto.InternalMessageInfo + +type isParameterRangeOneOf_ParameterRangeType interface { + isParameterRangeOneOf_ParameterRangeType() +} + +type ParameterRangeOneOf_ContinuousParameterRange struct { + ContinuousParameterRange *ContinuousParameterRange `protobuf:"bytes,1,opt,name=continuous_parameter_range,json=continuousParameterRange,proto3,oneof"` +} + +type ParameterRangeOneOf_IntegerParameterRange struct { + IntegerParameterRange *IntegerParameterRange `protobuf:"bytes,2,opt,name=integer_parameter_range,json=integerParameterRange,proto3,oneof"` +} + +type ParameterRangeOneOf_CategoricalParameterRange struct { + CategoricalParameterRange *CategoricalParameterRange `protobuf:"bytes,3,opt,name=categorical_parameter_range,json=categoricalParameterRange,proto3,oneof"` +} + +func (*ParameterRangeOneOf_ContinuousParameterRange) isParameterRangeOneOf_ParameterRangeType() {} + +func (*ParameterRangeOneOf_IntegerParameterRange) isParameterRangeOneOf_ParameterRangeType() {} + +func (*ParameterRangeOneOf_CategoricalParameterRange) isParameterRangeOneOf_ParameterRangeType() {} + +func (m *ParameterRangeOneOf) GetParameterRangeType() isParameterRangeOneOf_ParameterRangeType { + if m != nil { + return m.ParameterRangeType + } + return nil +} + +func (m *ParameterRangeOneOf) GetContinuousParameterRange() *ContinuousParameterRange { + if x, ok := m.GetParameterRangeType().(*ParameterRangeOneOf_ContinuousParameterRange); ok { + return x.ContinuousParameterRange + } + return nil +} + +func (m *ParameterRangeOneOf) GetIntegerParameterRange() *IntegerParameterRange { + if x, ok := m.GetParameterRangeType().(*ParameterRangeOneOf_IntegerParameterRange); ok { + return x.IntegerParameterRange + } + return nil +} + +func (m *ParameterRangeOneOf) GetCategoricalParameterRange() *CategoricalParameterRange { + if x, ok := m.GetParameterRangeType().(*ParameterRangeOneOf_CategoricalParameterRange); ok { + return x.CategoricalParameterRange + } + return nil +} + +// XXX_OneofWrappers is for the internal use of the proto package. +func (*ParameterRangeOneOf) XXX_OneofWrappers() []interface{} { + return []interface{}{ + (*ParameterRangeOneOf_ContinuousParameterRange)(nil), + (*ParameterRangeOneOf_IntegerParameterRange)(nil), + (*ParameterRangeOneOf_CategoricalParameterRange)(nil), + } +} + +// ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +type ParameterRanges struct { + ParameterRangeMap map[string]*ParameterRangeOneOf `protobuf:"bytes,1,rep,name=parameter_range_map,json=parameterRangeMap,proto3" json:"parameter_range_map,omitempty" protobuf_key:"bytes,1,opt,name=key,proto3" protobuf_val:"bytes,2,opt,name=value,proto3"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *ParameterRanges) Reset() { *m = ParameterRanges{} } +func (m *ParameterRanges) String() string { return proto.CompactTextString(m) } +func (*ParameterRanges) ProtoMessage() {} +func (*ParameterRanges) Descriptor() ([]byte, []int) { + return fileDescriptor_5f31fcc87eba0a70, []int{5} +} + +func (m *ParameterRanges) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_ParameterRanges.Unmarshal(m, b) +} +func (m *ParameterRanges) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_ParameterRanges.Marshal(b, m, deterministic) +} +func (m *ParameterRanges) XXX_Merge(src proto.Message) { + xxx_messageInfo_ParameterRanges.Merge(m, src) +} +func (m *ParameterRanges) XXX_Size() int { + return xxx_messageInfo_ParameterRanges.Size(m) +} +func (m *ParameterRanges) XXX_DiscardUnknown() { + xxx_messageInfo_ParameterRanges.DiscardUnknown(m) +} + +var xxx_messageInfo_ParameterRanges proto.InternalMessageInfo + +func (m *ParameterRanges) GetParameterRangeMap() map[string]*ParameterRangeOneOf { + if m != nil { + return m.ParameterRangeMap + } + return nil +} + +func init() { + proto.RegisterEnum("flyteidl.plugins.sagemaker.HyperparameterScalingType_Value", HyperparameterScalingType_Value_name, HyperparameterScalingType_Value_value) + proto.RegisterType((*HyperparameterScalingType)(nil), "flyteidl.plugins.sagemaker.HyperparameterScalingType") + proto.RegisterType((*ContinuousParameterRange)(nil), "flyteidl.plugins.sagemaker.ContinuousParameterRange") + proto.RegisterType((*IntegerParameterRange)(nil), "flyteidl.plugins.sagemaker.IntegerParameterRange") + proto.RegisterType((*CategoricalParameterRange)(nil), "flyteidl.plugins.sagemaker.CategoricalParameterRange") + proto.RegisterType((*ParameterRangeOneOf)(nil), "flyteidl.plugins.sagemaker.ParameterRangeOneOf") + proto.RegisterType((*ParameterRanges)(nil), "flyteidl.plugins.sagemaker.ParameterRanges") + proto.RegisterMapType((map[string]*ParameterRangeOneOf)(nil), "flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry") +} + +func init() { + proto.RegisterFile("flyteidl/plugins/sagemaker/parameter_ranges.proto", fileDescriptor_5f31fcc87eba0a70) +} + +var fileDescriptor_5f31fcc87eba0a70 = []byte{ + // 511 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0xbc, 0x54, 0x41, 0x6f, 0x12, 0x41, + 0x14, 0x66, 0x58, 0xdb, 0x94, 0x87, 0xb1, 0x38, 0xb5, 0x08, 0xf4, 0x42, 0x38, 0x71, 0x71, 0x37, + 0x05, 0x9b, 0x18, 0x3d, 0x01, 0x59, 0x85, 0xa4, 0x15, 0x33, 0xc5, 0x1e, 0x3c, 0x48, 0x86, 0x75, + 0x58, 0x27, 0xec, 0xce, 0x4e, 0x66, 0x67, 0xb5, 0xfb, 0x13, 0xfc, 0x39, 0x26, 0xfe, 0x3c, 0x0f, + 0x86, 0x81, 0xd2, 0x76, 0xcb, 0xa2, 0x27, 0x6f, 0x33, 0xf3, 0xde, 0x9b, 0xef, 0x7b, 0xef, 0xfb, + 0xf2, 0xe0, 0x74, 0x1e, 0xa4, 0x9a, 0xf1, 0x2f, 0x81, 0x23, 0x83, 0xc4, 0xe7, 0x22, 0x76, 0x62, + 0xea, 0xb3, 0x90, 0x2e, 0x98, 0x72, 0x24, 0x55, 0x34, 0x64, 0x9a, 0xa9, 0xa9, 0xa2, 0xc2, 0x67, + 0xb1, 0x2d, 0x55, 0xa4, 0x23, 0xdc, 0xb8, 0x29, 0xb1, 0xd7, 0x25, 0xf6, 0xa6, 0xa4, 0xe5, 0x41, + 0x7d, 0x98, 0x4a, 0xa6, 0x36, 0xa5, 0x97, 0x1e, 0x0d, 0xb8, 0xf0, 0x27, 0xa9, 0x64, 0xad, 0xb7, + 0xb0, 0x77, 0x45, 0x83, 0x84, 0xe1, 0x03, 0x78, 0xd4, 0xfb, 0x38, 0x19, 0x57, 0x0a, 0x18, 0x60, + 0xff, 0x7c, 0xf4, 0xde, 0xed, 0x91, 0x0a, 0xc2, 0x87, 0x50, 0x3e, 0x1f, 0xbf, 0xeb, 0x91, 0xd1, + 0x64, 0x78, 0x31, 0x1a, 0x54, 0x8a, 0xb8, 0x0a, 0x98, 0xb8, 0x57, 0x2e, 0xb9, 0x74, 0xef, 0xbe, + 0x5b, 0xad, 0x5f, 0x08, 0x6a, 0x83, 0x48, 0x68, 0x2e, 0x92, 0x28, 0x89, 0x3f, 0xdc, 0x40, 0x91, + 0x25, 0x49, 0x7c, 0x02, 0xa5, 0x90, 0x5e, 0x4f, 0xbf, 0x2d, 0x81, 0x6a, 0xa8, 0x89, 0xda, 0x88, + 0x1c, 0x84, 0xf4, 0x7a, 0x05, 0xbc, 0x0c, 0x72, 0xb1, 0x0e, 0x16, 0xd7, 0x41, 0x2e, 0x56, 0xc1, + 0xcf, 0xf0, 0x38, 0x5e, 0xb1, 0x9d, 0xea, 0x54, 0xb2, 0x9a, 0xd5, 0x44, 0xed, 0x27, 0x9d, 0x37, + 0x76, 0x7e, 0xbb, 0x76, 0x6e, 0xaf, 0xb6, 0xf9, 0x92, 0x94, 0xe3, 0x3b, 0xed, 0xff, 0x44, 0x70, + 0x3c, 0x12, 0x9a, 0xf9, 0x4c, 0xfd, 0x8d, 0xb3, 0xb5, 0x8b, 0xb3, 0xf5, 0x1f, 0x39, 0x77, 0xa1, + 0x3e, 0xa0, 0x9a, 0xf9, 0x91, 0xe2, 0x1e, 0x0d, 0x32, 0xb4, 0xab, 0xb0, 0x6f, 0x58, 0xc5, 0x35, + 0xd4, 0xb4, 0xda, 0x25, 0xb2, 0xbe, 0xb5, 0x7e, 0x58, 0x70, 0x74, 0x3f, 0x75, 0x2c, 0xd8, 0x78, + 0x8e, 0x35, 0x34, 0xbc, 0x8d, 0x6c, 0xd3, 0x8c, 0xbb, 0x4c, 0xdf, 0xe5, 0xce, 0xcb, 0x5d, 0xd4, + 0xf3, 0x44, 0x1f, 0x16, 0x48, 0xcd, 0xcb, 0x33, 0xc4, 0x02, 0x9e, 0xf3, 0xd5, 0xd4, 0x1f, 0x40, + 0x16, 0x0d, 0xe4, 0xe9, 0x2e, 0xc8, 0xad, 0x82, 0x0d, 0x0b, 0xe4, 0x98, 0x6f, 0x55, 0xf2, 0x3b, + 0x9c, 0x78, 0xb7, 0xf3, 0x7a, 0x00, 0x68, 0x19, 0xc0, 0xb3, 0x9d, 0x3d, 0xe6, 0x8d, 0x7b, 0x58, + 0x20, 0x75, 0x2f, 0x2f, 0xd8, 0xaf, 0xc2, 0xb3, 0x0c, 0x98, 0x31, 0x44, 0xeb, 0x37, 0x82, 0xc3, + 0xfb, 0xa9, 0x31, 0x56, 0x70, 0x94, 0xcd, 0x0d, 0xa9, 0x34, 0x22, 0x96, 0x3b, 0xfd, 0x5d, 0xe4, + 0x32, 0x3f, 0x65, 0xee, 0x17, 0x54, 0xba, 0x42, 0xab, 0x94, 0x3c, 0x95, 0xd9, 0xf7, 0x46, 0x02, + 0xd5, 0xed, 0xc9, 0xb8, 0x02, 0xd6, 0x82, 0xa5, 0x46, 0xfe, 0x12, 0x59, 0x1e, 0xb1, 0x0b, 0x7b, + 0xb7, 0x6e, 0x2f, 0x77, 0x9c, 0x7f, 0x67, 0x64, 0x7c, 0x46, 0x56, 0xd5, 0xaf, 0x8b, 0xaf, 0x50, + 0xff, 0xec, 0x53, 0xd7, 0xe7, 0xfa, 0x6b, 0x32, 0xb3, 0xbd, 0x28, 0x74, 0x82, 0x74, 0xae, 0x9d, + 0xcd, 0xc2, 0xf3, 0x99, 0x70, 0xe4, 0xec, 0x85, 0x1f, 0x39, 0xd9, 0x1d, 0x38, 0xdb, 0x37, 0x9b, + 0xae, 0xfb, 0x27, 0x00, 0x00, 0xff, 0xff, 0xdc, 0x29, 0x92, 0xb6, 0x1e, 0x05, 0x00, 0x00, +} diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.validate.go b/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.validate.go new file mode 100644 index 000000000..309486134 --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/parameter_ranges.pb.validate.go @@ -0,0 +1,506 @@ +// Code generated by protoc-gen-validate. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +package plugins + +import ( + "bytes" + "errors" + "fmt" + "net" + "net/mail" + "net/url" + "regexp" + "strings" + "time" + "unicode/utf8" + + "github.com/golang/protobuf/ptypes" +) + +// ensure the imports are used +var ( + _ = bytes.MinRead + _ = errors.New("") + _ = fmt.Print + _ = utf8.UTFMax + _ = (*regexp.Regexp)(nil) + _ = (*strings.Reader)(nil) + _ = net.IPv4len + _ = time.Duration(0) + _ = (*url.URL)(nil) + _ = (*mail.Address)(nil) + _ = ptypes.DynamicAny{} +) + +// define the regex for a UUID once up-front +var _parameter_ranges_uuidPattern = regexp.MustCompile("^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$") + +// Validate checks the field values on HyperparameterScalingType with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *HyperparameterScalingType) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// HyperparameterScalingTypeValidationError is the validation error returned by +// HyperparameterScalingType.Validate if the designated constraints aren't met. +type HyperparameterScalingTypeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e HyperparameterScalingTypeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e HyperparameterScalingTypeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e HyperparameterScalingTypeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e HyperparameterScalingTypeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e HyperparameterScalingTypeValidationError) ErrorName() string { + return "HyperparameterScalingTypeValidationError" +} + +// Error satisfies the builtin error interface +func (e HyperparameterScalingTypeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sHyperparameterScalingType.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = HyperparameterScalingTypeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = HyperparameterScalingTypeValidationError{} + +// Validate checks the field values on ContinuousParameterRange with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *ContinuousParameterRange) Validate() error { + if m == nil { + return nil + } + + // no validation rules for MaxValue + + // no validation rules for MinValue + + // no validation rules for ScalingType + + return nil +} + +// ContinuousParameterRangeValidationError is the validation error returned by +// ContinuousParameterRange.Validate if the designated constraints aren't met. +type ContinuousParameterRangeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e ContinuousParameterRangeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e ContinuousParameterRangeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e ContinuousParameterRangeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e ContinuousParameterRangeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e ContinuousParameterRangeValidationError) ErrorName() string { + return "ContinuousParameterRangeValidationError" +} + +// Error satisfies the builtin error interface +func (e ContinuousParameterRangeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sContinuousParameterRange.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = ContinuousParameterRangeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = ContinuousParameterRangeValidationError{} + +// Validate checks the field values on IntegerParameterRange with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *IntegerParameterRange) Validate() error { + if m == nil { + return nil + } + + // no validation rules for MaxValue + + // no validation rules for MinValue + + // no validation rules for ScalingType + + return nil +} + +// IntegerParameterRangeValidationError is the validation error returned by +// IntegerParameterRange.Validate if the designated constraints aren't met. +type IntegerParameterRangeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e IntegerParameterRangeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e IntegerParameterRangeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e IntegerParameterRangeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e IntegerParameterRangeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e IntegerParameterRangeValidationError) ErrorName() string { + return "IntegerParameterRangeValidationError" +} + +// Error satisfies the builtin error interface +func (e IntegerParameterRangeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sIntegerParameterRange.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = IntegerParameterRangeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = IntegerParameterRangeValidationError{} + +// Validate checks the field values on CategoricalParameterRange with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *CategoricalParameterRange) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// CategoricalParameterRangeValidationError is the validation error returned by +// CategoricalParameterRange.Validate if the designated constraints aren't met. +type CategoricalParameterRangeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e CategoricalParameterRangeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e CategoricalParameterRangeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e CategoricalParameterRangeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e CategoricalParameterRangeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e CategoricalParameterRangeValidationError) ErrorName() string { + return "CategoricalParameterRangeValidationError" +} + +// Error satisfies the builtin error interface +func (e CategoricalParameterRangeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sCategoricalParameterRange.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = CategoricalParameterRangeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = CategoricalParameterRangeValidationError{} + +// Validate checks the field values on ParameterRangeOneOf with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *ParameterRangeOneOf) Validate() error { + if m == nil { + return nil + } + + switch m.ParameterRangeType.(type) { + + case *ParameterRangeOneOf_ContinuousParameterRange: + + if v, ok := interface{}(m.GetContinuousParameterRange()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ParameterRangeOneOfValidationError{ + field: "ContinuousParameterRange", + reason: "embedded message failed validation", + cause: err, + } + } + } + + case *ParameterRangeOneOf_IntegerParameterRange: + + if v, ok := interface{}(m.GetIntegerParameterRange()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ParameterRangeOneOfValidationError{ + field: "IntegerParameterRange", + reason: "embedded message failed validation", + cause: err, + } + } + } + + case *ParameterRangeOneOf_CategoricalParameterRange: + + if v, ok := interface{}(m.GetCategoricalParameterRange()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ParameterRangeOneOfValidationError{ + field: "CategoricalParameterRange", + reason: "embedded message failed validation", + cause: err, + } + } + } + + } + + return nil +} + +// ParameterRangeOneOfValidationError is the validation error returned by +// ParameterRangeOneOf.Validate if the designated constraints aren't met. +type ParameterRangeOneOfValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e ParameterRangeOneOfValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e ParameterRangeOneOfValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e ParameterRangeOneOfValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e ParameterRangeOneOfValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e ParameterRangeOneOfValidationError) ErrorName() string { + return "ParameterRangeOneOfValidationError" +} + +// Error satisfies the builtin error interface +func (e ParameterRangeOneOfValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sParameterRangeOneOf.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = ParameterRangeOneOfValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = ParameterRangeOneOfValidationError{} + +// Validate checks the field values on ParameterRanges with the rules defined +// in the proto definition for this message. If any rules are violated, an +// error is returned. +func (m *ParameterRanges) Validate() error { + if m == nil { + return nil + } + + for key, val := range m.GetParameterRangeMap() { + _ = val + + // no validation rules for ParameterRangeMap[key] + + if v, ok := interface{}(val).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return ParameterRangesValidationError{ + field: fmt.Sprintf("ParameterRangeMap[%v]", key), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } + + return nil +} + +// ParameterRangesValidationError is the validation error returned by +// ParameterRanges.Validate if the designated constraints aren't met. +type ParameterRangesValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e ParameterRangesValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e ParameterRangesValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e ParameterRangesValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e ParameterRangesValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e ParameterRangesValidationError) ErrorName() string { return "ParameterRangesValidationError" } + +// Error satisfies the builtin error interface +func (e ParameterRangesValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sParameterRanges.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = ParameterRangesValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = ParameterRangesValidationError{} diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.go b/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.go new file mode 100644 index 000000000..865ea9a67 --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.go @@ -0,0 +1,509 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/training_job.proto + +package plugins + +import ( + fmt "fmt" + proto "github.com/golang/protobuf/proto" + _ "github.com/golang/protobuf/ptypes/duration" + math "math" +) + +// Reference imports to suppress errors if they are not otherwise used. +var _ = proto.Marshal +var _ = fmt.Errorf +var _ = math.Inf + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the proto package it is being compiled against. +// A compilation error at this line likely means your copy of the +// proto package needs to be updated. +const _ = proto.ProtoPackageIsVersion3 // please upgrade the proto package + +type InputMode_Value int32 + +const ( + InputMode_FILE InputMode_Value = 0 + InputMode_PIPE InputMode_Value = 1 +) + +var InputMode_Value_name = map[int32]string{ + 0: "FILE", + 1: "PIPE", +} + +var InputMode_Value_value = map[string]int32{ + "FILE": 0, + "PIPE": 1, +} + +func (x InputMode_Value) String() string { + return proto.EnumName(InputMode_Value_name, int32(x)) +} + +func (InputMode_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{0, 0} +} + +type AlgorithmName_Value int32 + +const ( + AlgorithmName_CUSTOM AlgorithmName_Value = 0 + AlgorithmName_XGBOOST AlgorithmName_Value = 1 +) + +var AlgorithmName_Value_name = map[int32]string{ + 0: "CUSTOM", + 1: "XGBOOST", +} + +var AlgorithmName_Value_value = map[string]int32{ + "CUSTOM": 0, + "XGBOOST": 1, +} + +func (x AlgorithmName_Value) String() string { + return proto.EnumName(AlgorithmName_Value_name, int32(x)) +} + +func (AlgorithmName_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{1, 0} +} + +type InputContentType_Value int32 + +const ( + InputContentType_TEXT_CSV InputContentType_Value = 0 +) + +var InputContentType_Value_name = map[int32]string{ + 0: "TEXT_CSV", +} + +var InputContentType_Value_value = map[string]int32{ + "TEXT_CSV": 0, +} + +func (x InputContentType_Value) String() string { + return proto.EnumName(InputContentType_Value_name, int32(x)) +} + +func (InputContentType_Value) EnumDescriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{2, 0} +} + +// The input mode that the algorithm supports. When using the File input mode, SageMaker downloads +// the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker +// volume for training container. When using Pipe input mode, Amazon SageMaker streams data directly +// from S3 to the container. +// See: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html +// For the input modes that different SageMaker algorithms support, see: +// https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html +type InputMode struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *InputMode) Reset() { *m = InputMode{} } +func (m *InputMode) String() string { return proto.CompactTextString(m) } +func (*InputMode) ProtoMessage() {} +func (*InputMode) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{0} +} + +func (m *InputMode) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_InputMode.Unmarshal(m, b) +} +func (m *InputMode) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_InputMode.Marshal(b, m, deterministic) +} +func (m *InputMode) XXX_Merge(src proto.Message) { + xxx_messageInfo_InputMode.Merge(m, src) +} +func (m *InputMode) XXX_Size() int { + return xxx_messageInfo_InputMode.Size(m) +} +func (m *InputMode) XXX_DiscardUnknown() { + xxx_messageInfo_InputMode.DiscardUnknown(m) +} + +var xxx_messageInfo_InputMode proto.InternalMessageInfo + +// The algorithm name is used for deciding which pre-built image to point to. +// This is only required for use cases where SageMaker's built-in algorithm mode is used. +// While we currently only support a subset of the algorithms, more will be added to the list. +// See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html +type AlgorithmName struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *AlgorithmName) Reset() { *m = AlgorithmName{} } +func (m *AlgorithmName) String() string { return proto.CompactTextString(m) } +func (*AlgorithmName) ProtoMessage() {} +func (*AlgorithmName) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{1} +} + +func (m *AlgorithmName) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_AlgorithmName.Unmarshal(m, b) +} +func (m *AlgorithmName) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_AlgorithmName.Marshal(b, m, deterministic) +} +func (m *AlgorithmName) XXX_Merge(src proto.Message) { + xxx_messageInfo_AlgorithmName.Merge(m, src) +} +func (m *AlgorithmName) XXX_Size() int { + return xxx_messageInfo_AlgorithmName.Size(m) +} +func (m *AlgorithmName) XXX_DiscardUnknown() { + xxx_messageInfo_AlgorithmName.DiscardUnknown(m) +} + +var xxx_messageInfo_AlgorithmName proto.InternalMessageInfo + +// Specifies the type of file for input data. Different SageMaker built-in algorithms require different file types of input data +// See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html +// https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html +type InputContentType struct { + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *InputContentType) Reset() { *m = InputContentType{} } +func (m *InputContentType) String() string { return proto.CompactTextString(m) } +func (*InputContentType) ProtoMessage() {} +func (*InputContentType) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{2} +} + +func (m *InputContentType) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_InputContentType.Unmarshal(m, b) +} +func (m *InputContentType) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_InputContentType.Marshal(b, m, deterministic) +} +func (m *InputContentType) XXX_Merge(src proto.Message) { + xxx_messageInfo_InputContentType.Merge(m, src) +} +func (m *InputContentType) XXX_Size() int { + return xxx_messageInfo_InputContentType.Size(m) +} +func (m *InputContentType) XXX_DiscardUnknown() { + xxx_messageInfo_InputContentType.DiscardUnknown(m) +} + +var xxx_messageInfo_InputContentType proto.InternalMessageInfo + +// Specifies a metric that the training algorithm writes to stderr or stdout. +// This object is a pass-through. +// See this for details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MetricDefinition.html +type MetricDefinition struct { + // User-defined name of the metric + Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"` + // SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics + Regex string `protobuf:"bytes,2,opt,name=regex,proto3" json:"regex,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *MetricDefinition) Reset() { *m = MetricDefinition{} } +func (m *MetricDefinition) String() string { return proto.CompactTextString(m) } +func (*MetricDefinition) ProtoMessage() {} +func (*MetricDefinition) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{3} +} + +func (m *MetricDefinition) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_MetricDefinition.Unmarshal(m, b) +} +func (m *MetricDefinition) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_MetricDefinition.Marshal(b, m, deterministic) +} +func (m *MetricDefinition) XXX_Merge(src proto.Message) { + xxx_messageInfo_MetricDefinition.Merge(m, src) +} +func (m *MetricDefinition) XXX_Size() int { + return xxx_messageInfo_MetricDefinition.Size(m) +} +func (m *MetricDefinition) XXX_DiscardUnknown() { + xxx_messageInfo_MetricDefinition.DiscardUnknown(m) +} + +var xxx_messageInfo_MetricDefinition proto.InternalMessageInfo + +func (m *MetricDefinition) GetName() string { + if m != nil { + return m.Name + } + return "" +} + +func (m *MetricDefinition) GetRegex() string { + if m != nil { + return m.Regex + } + return "" +} + +// Specifies the training algorithm to be used in the training job +// This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify +// TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm +// name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to +// CUSTOM. In this case, the value of the algorithm_version field has no effect +// For pass-through use cases: refer to this AWS official document for more details +// https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html +type AlgorithmSpecification struct { + // The input mode can be either PIPE or FILE + InputMode InputMode_Value `protobuf:"varint,1,opt,name=input_mode,json=inputMode,proto3,enum=flyteidl.plugins.sagemaker.InputMode_Value" json:"input_mode,omitempty"` + // The algorithm name is used for deciding which pre-built image to point to + AlgorithmName AlgorithmName_Value `protobuf:"varint,2,opt,name=algorithm_name,json=algorithmName,proto3,enum=flyteidl.plugins.sagemaker.AlgorithmName_Value" json:"algorithm_name,omitempty"` + // The algorithm version field is used for deciding which pre-built image to point to + // This is only needed for use cases where SageMaker's built-in algorithm mode is chosen + AlgorithmVersion string `protobuf:"bytes,3,opt,name=algorithm_version,json=algorithmVersion,proto3" json:"algorithm_version,omitempty"` + // A list of metric definitions for SageMaker to evaluate/track on the progress of the training job + // See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html + // and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html + MetricDefinitions []*MetricDefinition `protobuf:"bytes,4,rep,name=metric_definitions,json=metricDefinitions,proto3" json:"metric_definitions,omitempty"` + // The content type of the input + // See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html + // https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html + InputContentType InputContentType_Value `protobuf:"varint,5,opt,name=input_content_type,json=inputContentType,proto3,enum=flyteidl.plugins.sagemaker.InputContentType_Value" json:"input_content_type,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *AlgorithmSpecification) Reset() { *m = AlgorithmSpecification{} } +func (m *AlgorithmSpecification) String() string { return proto.CompactTextString(m) } +func (*AlgorithmSpecification) ProtoMessage() {} +func (*AlgorithmSpecification) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{4} +} + +func (m *AlgorithmSpecification) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_AlgorithmSpecification.Unmarshal(m, b) +} +func (m *AlgorithmSpecification) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_AlgorithmSpecification.Marshal(b, m, deterministic) +} +func (m *AlgorithmSpecification) XXX_Merge(src proto.Message) { + xxx_messageInfo_AlgorithmSpecification.Merge(m, src) +} +func (m *AlgorithmSpecification) XXX_Size() int { + return xxx_messageInfo_AlgorithmSpecification.Size(m) +} +func (m *AlgorithmSpecification) XXX_DiscardUnknown() { + xxx_messageInfo_AlgorithmSpecification.DiscardUnknown(m) +} + +var xxx_messageInfo_AlgorithmSpecification proto.InternalMessageInfo + +func (m *AlgorithmSpecification) GetInputMode() InputMode_Value { + if m != nil { + return m.InputMode + } + return InputMode_FILE +} + +func (m *AlgorithmSpecification) GetAlgorithmName() AlgorithmName_Value { + if m != nil { + return m.AlgorithmName + } + return AlgorithmName_CUSTOM +} + +func (m *AlgorithmSpecification) GetAlgorithmVersion() string { + if m != nil { + return m.AlgorithmVersion + } + return "" +} + +func (m *AlgorithmSpecification) GetMetricDefinitions() []*MetricDefinition { + if m != nil { + return m.MetricDefinitions + } + return nil +} + +func (m *AlgorithmSpecification) GetInputContentType() InputContentType_Value { + if m != nil { + return m.InputContentType + } + return InputContentType_TEXT_CSV +} + +// TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the +// number of instances to launch, and the size of the ML storage volume the user wants to provision +// Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html +type TrainingJobResourceConfig struct { + // The number of ML compute instances to use. For distributed training, provide a value greater than 1. + InstanceCount int64 `protobuf:"varint,1,opt,name=instance_count,json=instanceCount,proto3" json:"instance_count,omitempty"` + // The ML compute instance type + InstanceType string `protobuf:"bytes,2,opt,name=instance_type,json=instanceType,proto3" json:"instance_type,omitempty"` + // The size of the ML storage volume that you want to provision. + VolumeSizeInGb int64 `protobuf:"varint,3,opt,name=volume_size_in_gb,json=volumeSizeInGb,proto3" json:"volume_size_in_gb,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *TrainingJobResourceConfig) Reset() { *m = TrainingJobResourceConfig{} } +func (m *TrainingJobResourceConfig) String() string { return proto.CompactTextString(m) } +func (*TrainingJobResourceConfig) ProtoMessage() {} +func (*TrainingJobResourceConfig) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{5} +} + +func (m *TrainingJobResourceConfig) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_TrainingJobResourceConfig.Unmarshal(m, b) +} +func (m *TrainingJobResourceConfig) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_TrainingJobResourceConfig.Marshal(b, m, deterministic) +} +func (m *TrainingJobResourceConfig) XXX_Merge(src proto.Message) { + xxx_messageInfo_TrainingJobResourceConfig.Merge(m, src) +} +func (m *TrainingJobResourceConfig) XXX_Size() int { + return xxx_messageInfo_TrainingJobResourceConfig.Size(m) +} +func (m *TrainingJobResourceConfig) XXX_DiscardUnknown() { + xxx_messageInfo_TrainingJobResourceConfig.DiscardUnknown(m) +} + +var xxx_messageInfo_TrainingJobResourceConfig proto.InternalMessageInfo + +func (m *TrainingJobResourceConfig) GetInstanceCount() int64 { + if m != nil { + return m.InstanceCount + } + return 0 +} + +func (m *TrainingJobResourceConfig) GetInstanceType() string { + if m != nil { + return m.InstanceType + } + return "" +} + +func (m *TrainingJobResourceConfig) GetVolumeSizeInGb() int64 { + if m != nil { + return m.VolumeSizeInGb + } + return 0 +} + +// The spec of a training job. This is mostly a pass-through object +// https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html +type TrainingJob struct { + AlgorithmSpecification *AlgorithmSpecification `protobuf:"bytes,1,opt,name=algorithm_specification,json=algorithmSpecification,proto3" json:"algorithm_specification,omitempty"` + TrainingJobResourceConfig *TrainingJobResourceConfig `protobuf:"bytes,2,opt,name=training_job_resource_config,json=trainingJobResourceConfig,proto3" json:"training_job_resource_config,omitempty"` + XXX_NoUnkeyedLiteral struct{} `json:"-"` + XXX_unrecognized []byte `json:"-"` + XXX_sizecache int32 `json:"-"` +} + +func (m *TrainingJob) Reset() { *m = TrainingJob{} } +func (m *TrainingJob) String() string { return proto.CompactTextString(m) } +func (*TrainingJob) ProtoMessage() {} +func (*TrainingJob) Descriptor() ([]byte, []int) { + return fileDescriptor_6a68f64d8fd9fe30, []int{6} +} + +func (m *TrainingJob) XXX_Unmarshal(b []byte) error { + return xxx_messageInfo_TrainingJob.Unmarshal(m, b) +} +func (m *TrainingJob) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) { + return xxx_messageInfo_TrainingJob.Marshal(b, m, deterministic) +} +func (m *TrainingJob) XXX_Merge(src proto.Message) { + xxx_messageInfo_TrainingJob.Merge(m, src) +} +func (m *TrainingJob) XXX_Size() int { + return xxx_messageInfo_TrainingJob.Size(m) +} +func (m *TrainingJob) XXX_DiscardUnknown() { + xxx_messageInfo_TrainingJob.DiscardUnknown(m) +} + +var xxx_messageInfo_TrainingJob proto.InternalMessageInfo + +func (m *TrainingJob) GetAlgorithmSpecification() *AlgorithmSpecification { + if m != nil { + return m.AlgorithmSpecification + } + return nil +} + +func (m *TrainingJob) GetTrainingJobResourceConfig() *TrainingJobResourceConfig { + if m != nil { + return m.TrainingJobResourceConfig + } + return nil +} + +func init() { + proto.RegisterEnum("flyteidl.plugins.sagemaker.InputMode_Value", InputMode_Value_name, InputMode_Value_value) + proto.RegisterEnum("flyteidl.plugins.sagemaker.AlgorithmName_Value", AlgorithmName_Value_name, AlgorithmName_Value_value) + proto.RegisterEnum("flyteidl.plugins.sagemaker.InputContentType_Value", InputContentType_Value_name, InputContentType_Value_value) + proto.RegisterType((*InputMode)(nil), "flyteidl.plugins.sagemaker.InputMode") + proto.RegisterType((*AlgorithmName)(nil), "flyteidl.plugins.sagemaker.AlgorithmName") + proto.RegisterType((*InputContentType)(nil), "flyteidl.plugins.sagemaker.InputContentType") + proto.RegisterType((*MetricDefinition)(nil), "flyteidl.plugins.sagemaker.MetricDefinition") + proto.RegisterType((*AlgorithmSpecification)(nil), "flyteidl.plugins.sagemaker.AlgorithmSpecification") + proto.RegisterType((*TrainingJobResourceConfig)(nil), "flyteidl.plugins.sagemaker.TrainingJobResourceConfig") + proto.RegisterType((*TrainingJob)(nil), "flyteidl.plugins.sagemaker.TrainingJob") +} + +func init() { + proto.RegisterFile("flyteidl/plugins/sagemaker/training_job.proto", fileDescriptor_6a68f64d8fd9fe30) +} + +var fileDescriptor_6a68f64d8fd9fe30 = []byte{ + // 592 bytes of a gzipped FileDescriptorProto + 0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x84, 0x94, 0xd1, 0x4e, 0xdb, 0x3c, + 0x14, 0xc7, 0x09, 0x05, 0x3e, 0x38, 0x85, 0x2a, 0x58, 0xdf, 0x58, 0x61, 0xd3, 0x84, 0x32, 0x4d, + 0x02, 0x31, 0x12, 0xad, 0x88, 0xbb, 0xdd, 0x8c, 0x8e, 0xa1, 0xa2, 0x31, 0x50, 0x9a, 0x55, 0x68, + 0xbb, 0xc8, 0x92, 0xf4, 0xc4, 0x78, 0x24, 0x76, 0x94, 0x38, 0x68, 0xe5, 0x31, 0xf6, 0x14, 0x7b, + 0xbe, 0x3d, 0xc1, 0x14, 0xa7, 0x09, 0x5d, 0x45, 0xbb, 0x3b, 0xe7, 0xef, 0xe3, 0xe3, 0xff, 0xf9, + 0x9d, 0x13, 0xc3, 0x61, 0x18, 0x8d, 0x24, 0xb2, 0x61, 0x64, 0x25, 0x51, 0x4e, 0x19, 0xcf, 0xac, + 0xcc, 0xa3, 0x18, 0x7b, 0xb7, 0x98, 0x5a, 0x32, 0xf5, 0x18, 0x67, 0x9c, 0xba, 0xdf, 0x85, 0x6f, + 0x26, 0xa9, 0x90, 0x82, 0xec, 0x54, 0xe1, 0xe6, 0x38, 0xdc, 0xac, 0xc3, 0x77, 0x5e, 0x50, 0x21, + 0x68, 0x84, 0x96, 0x8a, 0xf4, 0xf3, 0xd0, 0x1a, 0xe6, 0xa9, 0x27, 0x99, 0xe0, 0xe5, 0x59, 0x63, + 0x0f, 0xd6, 0x7a, 0x3c, 0xc9, 0xe5, 0x85, 0x18, 0xa2, 0xf1, 0x0c, 0x96, 0x07, 0x5e, 0x94, 0x23, + 0x59, 0x85, 0xa5, 0x0f, 0xbd, 0x8f, 0xa7, 0xfa, 0x42, 0xb1, 0xba, 0xea, 0x5d, 0x9d, 0xea, 0x9a, + 0xf1, 0x06, 0x36, 0xde, 0x45, 0x54, 0xa4, 0x4c, 0xde, 0xc4, 0x9f, 0xbc, 0x18, 0x8d, 0xdd, 0x2a, + 0x1a, 0x60, 0xa5, 0xfb, 0xb9, 0xef, 0x5c, 0x5e, 0xe8, 0x0b, 0xa4, 0x09, 0xff, 0x5d, 0x9f, 0x9d, + 0x5c, 0x5e, 0xf6, 0x1d, 0x5d, 0x33, 0xf6, 0x41, 0x57, 0xc9, 0xbb, 0x82, 0x4b, 0xe4, 0xd2, 0x19, + 0x25, 0x68, 0x3c, 0xa9, 0x4e, 0xad, 0xc3, 0xaa, 0x73, 0x7a, 0xed, 0xb8, 0xdd, 0xfe, 0x40, 0x5f, + 0x30, 0xde, 0x82, 0x7e, 0x81, 0x32, 0x65, 0xc1, 0x7b, 0x0c, 0x19, 0x67, 0x85, 0x43, 0x42, 0x60, + 0x89, 0x7b, 0x31, 0xb6, 0xb5, 0x5d, 0x6d, 0x6f, 0xcd, 0x56, 0x6b, 0xf2, 0x3f, 0x2c, 0xa7, 0x48, + 0xf1, 0x47, 0x7b, 0x51, 0x89, 0xe5, 0x87, 0xf1, 0xab, 0x01, 0x5b, 0xb5, 0xb9, 0x7e, 0x82, 0x01, + 0x0b, 0x59, 0xa0, 0xca, 0x24, 0xe7, 0x00, 0xac, 0xf0, 0xe0, 0xc6, 0x62, 0x58, 0xa6, 0x6a, 0x75, + 0x0e, 0xcc, 0xd9, 0xc4, 0xcc, 0x1a, 0x87, 0xa9, 0x7c, 0xda, 0x6b, 0xac, 0x12, 0xc8, 0x00, 0x5a, + 0x5e, 0x75, 0x8b, 0xab, 0xac, 0x2d, 0xaa, 0x7c, 0xd6, 0xbc, 0x7c, 0x7f, 0x41, 0x1b, 0xe7, 0xdc, + 0xf0, 0x26, 0x45, 0x72, 0x00, 0x9b, 0x0f, 0x79, 0xef, 0x30, 0xcd, 0x98, 0xe0, 0xed, 0x86, 0x2a, + 0x50, 0xaf, 0x37, 0x06, 0xa5, 0x4e, 0xbe, 0x02, 0x89, 0x15, 0x29, 0x77, 0x58, 0xa3, 0xca, 0xda, + 0x4b, 0xbb, 0x8d, 0xbd, 0x66, 0xe7, 0xf5, 0x3c, 0x23, 0xd3, 0x7c, 0xed, 0xcd, 0x78, 0x4a, 0xc9, + 0xc8, 0x37, 0x20, 0x25, 0xad, 0xa0, 0x6c, 0x99, 0x2b, 0x47, 0x09, 0xb6, 0x97, 0x55, 0x95, 0x9d, + 0x7f, 0x52, 0x9b, 0xe8, 0xf3, 0xb8, 0x50, 0x9d, 0x4d, 0xf7, 0xff, 0xa7, 0x06, 0xdb, 0xce, 0x78, + 0x86, 0xcf, 0x85, 0x6f, 0x63, 0x26, 0xf2, 0x34, 0xc0, 0xae, 0xe0, 0x21, 0xa3, 0xe4, 0x15, 0xb4, + 0x18, 0xcf, 0xa4, 0xc7, 0x03, 0x74, 0x03, 0x91, 0x73, 0xa9, 0x3a, 0xd6, 0xb0, 0x37, 0x2a, 0xb5, + 0x5b, 0x88, 0xe4, 0x25, 0xd4, 0x42, 0xe9, 0xb0, 0x9c, 0x86, 0xf5, 0x4a, 0x2c, 0x6e, 0x22, 0xfb, + 0xb0, 0x79, 0x27, 0xa2, 0x3c, 0x46, 0x37, 0x63, 0xf7, 0xe8, 0x32, 0xee, 0x52, 0x5f, 0x51, 0x6d, + 0xd8, 0xad, 0x72, 0xa3, 0xcf, 0xee, 0xb1, 0xc7, 0xcf, 0x7c, 0xe3, 0xb7, 0x06, 0xcd, 0x09, 0x53, + 0xe4, 0x16, 0x9e, 0x3e, 0x34, 0x24, 0x9b, 0x9c, 0x27, 0xe5, 0xa7, 0x39, 0x9f, 0xc5, 0xe3, 0x93, + 0x68, 0x6f, 0x79, 0x8f, 0x4f, 0xe8, 0x1d, 0x3c, 0x9f, 0xfc, 0xa9, 0xdd, 0x74, 0x8c, 0xa4, 0xe8, + 0x41, 0xc8, 0xa8, 0xaa, 0xad, 0xd9, 0x39, 0x9e, 0x77, 0xe3, 0x4c, 0xa0, 0xf6, 0xb6, 0x9c, 0xb5, + 0x75, 0x72, 0xfc, 0xe5, 0x88, 0x32, 0x79, 0x93, 0xfb, 0x66, 0x20, 0x62, 0x2b, 0x1a, 0x85, 0xd2, + 0xaa, 0xdf, 0x1d, 0x8a, 0xdc, 0x4a, 0xfc, 0x43, 0x2a, 0xac, 0xe9, 0xa7, 0xc8, 0x5f, 0x51, 0x0f, + 0xc7, 0xd1, 0x9f, 0x00, 0x00, 0x00, 0xff, 0xff, 0xc4, 0x72, 0xc9, 0x58, 0xa5, 0x04, 0x00, 0x00, +} diff --git a/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.validate.go b/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.validate.go new file mode 100644 index 000000000..a1ceaa438 --- /dev/null +++ b/gen/pb-go/flyteidl/plugins/sagemaker/training_job.pb.validate.go @@ -0,0 +1,548 @@ +// Code generated by protoc-gen-validate. DO NOT EDIT. +// source: flyteidl/plugins/sagemaker/training_job.proto + +package plugins + +import ( + "bytes" + "errors" + "fmt" + "net" + "net/mail" + "net/url" + "regexp" + "strings" + "time" + "unicode/utf8" + + "github.com/golang/protobuf/ptypes" +) + +// ensure the imports are used +var ( + _ = bytes.MinRead + _ = errors.New("") + _ = fmt.Print + _ = utf8.UTFMax + _ = (*regexp.Regexp)(nil) + _ = (*strings.Reader)(nil) + _ = net.IPv4len + _ = time.Duration(0) + _ = (*url.URL)(nil) + _ = (*mail.Address)(nil) + _ = ptypes.DynamicAny{} +) + +// define the regex for a UUID once up-front +var _training_job_uuidPattern = regexp.MustCompile("^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$") + +// Validate checks the field values on InputMode with the rules defined in the +// proto definition for this message. If any rules are violated, an error is returned. +func (m *InputMode) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// InputModeValidationError is the validation error returned by +// InputMode.Validate if the designated constraints aren't met. +type InputModeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e InputModeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e InputModeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e InputModeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e InputModeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e InputModeValidationError) ErrorName() string { return "InputModeValidationError" } + +// Error satisfies the builtin error interface +func (e InputModeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sInputMode.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = InputModeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = InputModeValidationError{} + +// Validate checks the field values on AlgorithmName with the rules defined in +// the proto definition for this message. If any rules are violated, an error +// is returned. +func (m *AlgorithmName) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// AlgorithmNameValidationError is the validation error returned by +// AlgorithmName.Validate if the designated constraints aren't met. +type AlgorithmNameValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e AlgorithmNameValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e AlgorithmNameValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e AlgorithmNameValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e AlgorithmNameValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e AlgorithmNameValidationError) ErrorName() string { return "AlgorithmNameValidationError" } + +// Error satisfies the builtin error interface +func (e AlgorithmNameValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sAlgorithmName.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = AlgorithmNameValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = AlgorithmNameValidationError{} + +// Validate checks the field values on InputContentType with the rules defined +// in the proto definition for this message. If any rules are violated, an +// error is returned. +func (m *InputContentType) Validate() error { + if m == nil { + return nil + } + + return nil +} + +// InputContentTypeValidationError is the validation error returned by +// InputContentType.Validate if the designated constraints aren't met. +type InputContentTypeValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e InputContentTypeValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e InputContentTypeValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e InputContentTypeValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e InputContentTypeValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e InputContentTypeValidationError) ErrorName() string { return "InputContentTypeValidationError" } + +// Error satisfies the builtin error interface +func (e InputContentTypeValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sInputContentType.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = InputContentTypeValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = InputContentTypeValidationError{} + +// Validate checks the field values on MetricDefinition with the rules defined +// in the proto definition for this message. If any rules are violated, an +// error is returned. +func (m *MetricDefinition) Validate() error { + if m == nil { + return nil + } + + // no validation rules for Name + + // no validation rules for Regex + + return nil +} + +// MetricDefinitionValidationError is the validation error returned by +// MetricDefinition.Validate if the designated constraints aren't met. +type MetricDefinitionValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e MetricDefinitionValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e MetricDefinitionValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e MetricDefinitionValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e MetricDefinitionValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e MetricDefinitionValidationError) ErrorName() string { return "MetricDefinitionValidationError" } + +// Error satisfies the builtin error interface +func (e MetricDefinitionValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sMetricDefinition.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = MetricDefinitionValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = MetricDefinitionValidationError{} + +// Validate checks the field values on AlgorithmSpecification with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *AlgorithmSpecification) Validate() error { + if m == nil { + return nil + } + + // no validation rules for InputMode + + // no validation rules for AlgorithmName + + // no validation rules for AlgorithmVersion + + for idx, item := range m.GetMetricDefinitions() { + _, _ = idx, item + + if v, ok := interface{}(item).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return AlgorithmSpecificationValidationError{ + field: fmt.Sprintf("MetricDefinitions[%v]", idx), + reason: "embedded message failed validation", + cause: err, + } + } + } + + } + + // no validation rules for InputContentType + + return nil +} + +// AlgorithmSpecificationValidationError is the validation error returned by +// AlgorithmSpecification.Validate if the designated constraints aren't met. +type AlgorithmSpecificationValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e AlgorithmSpecificationValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e AlgorithmSpecificationValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e AlgorithmSpecificationValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e AlgorithmSpecificationValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e AlgorithmSpecificationValidationError) ErrorName() string { + return "AlgorithmSpecificationValidationError" +} + +// Error satisfies the builtin error interface +func (e AlgorithmSpecificationValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sAlgorithmSpecification.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = AlgorithmSpecificationValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = AlgorithmSpecificationValidationError{} + +// Validate checks the field values on TrainingJobResourceConfig with the rules +// defined in the proto definition for this message. If any rules are +// violated, an error is returned. +func (m *TrainingJobResourceConfig) Validate() error { + if m == nil { + return nil + } + + // no validation rules for InstanceCount + + // no validation rules for InstanceType + + // no validation rules for VolumeSizeInGb + + return nil +} + +// TrainingJobResourceConfigValidationError is the validation error returned by +// TrainingJobResourceConfig.Validate if the designated constraints aren't met. +type TrainingJobResourceConfigValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e TrainingJobResourceConfigValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e TrainingJobResourceConfigValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e TrainingJobResourceConfigValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e TrainingJobResourceConfigValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e TrainingJobResourceConfigValidationError) ErrorName() string { + return "TrainingJobResourceConfigValidationError" +} + +// Error satisfies the builtin error interface +func (e TrainingJobResourceConfigValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sTrainingJobResourceConfig.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = TrainingJobResourceConfigValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = TrainingJobResourceConfigValidationError{} + +// Validate checks the field values on TrainingJob with the rules defined in +// the proto definition for this message. If any rules are violated, an error +// is returned. +func (m *TrainingJob) Validate() error { + if m == nil { + return nil + } + + if v, ok := interface{}(m.GetAlgorithmSpecification()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return TrainingJobValidationError{ + field: "AlgorithmSpecification", + reason: "embedded message failed validation", + cause: err, + } + } + } + + if v, ok := interface{}(m.GetTrainingJobResourceConfig()).(interface{ Validate() error }); ok { + if err := v.Validate(); err != nil { + return TrainingJobValidationError{ + field: "TrainingJobResourceConfig", + reason: "embedded message failed validation", + cause: err, + } + } + } + + return nil +} + +// TrainingJobValidationError is the validation error returned by +// TrainingJob.Validate if the designated constraints aren't met. +type TrainingJobValidationError struct { + field string + reason string + cause error + key bool +} + +// Field function returns field value. +func (e TrainingJobValidationError) Field() string { return e.field } + +// Reason function returns reason value. +func (e TrainingJobValidationError) Reason() string { return e.reason } + +// Cause function returns cause value. +func (e TrainingJobValidationError) Cause() error { return e.cause } + +// Key function returns key value. +func (e TrainingJobValidationError) Key() bool { return e.key } + +// ErrorName returns error name. +func (e TrainingJobValidationError) ErrorName() string { return "TrainingJobValidationError" } + +// Error satisfies the builtin error interface +func (e TrainingJobValidationError) Error() string { + cause := "" + if e.cause != nil { + cause = fmt.Sprintf(" | caused by: %v", e.cause) + } + + key := "" + if e.key { + key = "key for " + } + + return fmt.Sprintf( + "invalid %sTrainingJob.%s: %s%s", + key, + e.field, + e.reason, + cause) +} + +var _ error = TrainingJobValidationError{} + +var _ interface { + Field() string + Reason() string + Key() bool + Cause() error + ErrorName() string +} = TrainingJobValidationError{} diff --git a/gen/pb-go/flyteidl/service/openapi.go b/gen/pb-go/flyteidl/service/openapi.go index 7e6cbd41b..8673e6a38 100644 --- a/gen/pb-go/flyteidl/service/openapi.go +++ b/gen/pb-go/flyteidl/service/openapi.go @@ -1,5 +1,6 @@ -// Package service Code generated by go-bindata. (@generated) DO NOT EDIT. -// sources: +// Code generated by go-bindata. (@generated) DO NOT EDIT. + + //Package service generated by go-bindata.// sources: // ../service/admin.swagger.json package service diff --git a/gen/pb-java/flyteidl/plugins/sagemaker/HyperparameterTuningJobOuterClass.java b/gen/pb-java/flyteidl/plugins/sagemaker/HyperparameterTuningJobOuterClass.java new file mode 100644 index 000000000..96c23333a --- /dev/null +++ b/gen/pb-java/flyteidl/plugins/sagemaker/HyperparameterTuningJobOuterClass.java @@ -0,0 +1,4496 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +package flyteidl.plugins.sagemaker; + +public final class HyperparameterTuningJobOuterClass { + private HyperparameterTuningJobOuterClass() {} + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistryLite registry) { + } + + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistry registry) { + registerAllExtensions( + (com.google.protobuf.ExtensionRegistryLite) registry); + } + public interface HyperparameterTuningJobOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + boolean hasTrainingJob(); + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getTrainingJob(); + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder getTrainingJobOrBuilder(); + + /** + *
+     * The maximum number of training jobs that an hpo job can launch. For resource limit purpose.
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+     * 
+ * + * int64 max_number_of_training_jobs = 2; + */ + long getMaxNumberOfTrainingJobs(); + + /** + *
+     * The maximum number of concurrent training job that an hpo job can launch
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+     * 
+ * + * int64 max_parallel_training_jobs = 3; + */ + long getMaxParallelTrainingJobs(); + } + /** + *
+   * A pass-through for SageMaker's hyperparameter tuning job
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningJob} + */ + public static final class HyperparameterTuningJob extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + HyperparameterTuningJobOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterTuningJob.newBuilder() to construct. + private HyperparameterTuningJob(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterTuningJob() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterTuningJob( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder subBuilder = null; + if (trainingJob_ != null) { + subBuilder = trainingJob_.toBuilder(); + } + trainingJob_ = input.readMessage(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom(trainingJob_); + trainingJob_ = subBuilder.buildPartial(); + } + + break; + } + case 16: { + + maxNumberOfTrainingJobs_ = input.readInt64(); + break; + } + case 24: { + + maxParallelTrainingJobs_ = input.readInt64(); + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.Builder.class); + } + + public static final int TRAINING_JOB_FIELD_NUMBER = 1; + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob trainingJob_; + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public boolean hasTrainingJob() { + return trainingJob_ != null; + } + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getTrainingJob() { + return trainingJob_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.getDefaultInstance() : trainingJob_; + } + /** + *
+     * The underlying training job that the hyperparameter tuning job will launch during the process
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder getTrainingJobOrBuilder() { + return getTrainingJob(); + } + + public static final int MAX_NUMBER_OF_TRAINING_JOBS_FIELD_NUMBER = 2; + private long maxNumberOfTrainingJobs_; + /** + *
+     * The maximum number of training jobs that an hpo job can launch. For resource limit purpose.
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+     * 
+ * + * int64 max_number_of_training_jobs = 2; + */ + public long getMaxNumberOfTrainingJobs() { + return maxNumberOfTrainingJobs_; + } + + public static final int MAX_PARALLEL_TRAINING_JOBS_FIELD_NUMBER = 3; + private long maxParallelTrainingJobs_; + /** + *
+     * The maximum number of concurrent training job that an hpo job can launch
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+     * 
+ * + * int64 max_parallel_training_jobs = 3; + */ + public long getMaxParallelTrainingJobs() { + return maxParallelTrainingJobs_; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (trainingJob_ != null) { + output.writeMessage(1, getTrainingJob()); + } + if (maxNumberOfTrainingJobs_ != 0L) { + output.writeInt64(2, maxNumberOfTrainingJobs_); + } + if (maxParallelTrainingJobs_ != 0L) { + output.writeInt64(3, maxParallelTrainingJobs_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (trainingJob_ != null) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(1, getTrainingJob()); + } + if (maxNumberOfTrainingJobs_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(2, maxNumberOfTrainingJobs_); + } + if (maxParallelTrainingJobs_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(3, maxParallelTrainingJobs_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob) obj; + + if (hasTrainingJob() != other.hasTrainingJob()) return false; + if (hasTrainingJob()) { + if (!getTrainingJob() + .equals(other.getTrainingJob())) return false; + } + if (getMaxNumberOfTrainingJobs() + != other.getMaxNumberOfTrainingJobs()) return false; + if (getMaxParallelTrainingJobs() + != other.getMaxParallelTrainingJobs()) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + if (hasTrainingJob()) { + hash = (37 * hash) + TRAINING_JOB_FIELD_NUMBER; + hash = (53 * hash) + getTrainingJob().hashCode(); + } + hash = (37 * hash) + MAX_NUMBER_OF_TRAINING_JOBS_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getMaxNumberOfTrainingJobs()); + hash = (37 * hash) + MAX_PARALLEL_TRAINING_JOBS_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getMaxParallelTrainingJobs()); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * A pass-through for SageMaker's hyperparameter tuning job
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningJob} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + if (trainingJobBuilder_ == null) { + trainingJob_ = null; + } else { + trainingJob_ = null; + trainingJobBuilder_ = null; + } + maxNumberOfTrainingJobs_ = 0L; + + maxParallelTrainingJobs_ = 0L; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob(this); + if (trainingJobBuilder_ == null) { + result.trainingJob_ = trainingJob_; + } else { + result.trainingJob_ = trainingJobBuilder_.build(); + } + result.maxNumberOfTrainingJobs_ = maxNumberOfTrainingJobs_; + result.maxParallelTrainingJobs_ = maxParallelTrainingJobs_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob.getDefaultInstance()) return this; + if (other.hasTrainingJob()) { + mergeTrainingJob(other.getTrainingJob()); + } + if (other.getMaxNumberOfTrainingJobs() != 0L) { + setMaxNumberOfTrainingJobs(other.getMaxNumberOfTrainingJobs()); + } + if (other.getMaxParallelTrainingJobs() != 0L) { + setMaxParallelTrainingJobs(other.getMaxParallelTrainingJobs()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob trainingJob_; + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder> trainingJobBuilder_; + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public boolean hasTrainingJob() { + return trainingJobBuilder_ != null || trainingJob_ != null; + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getTrainingJob() { + if (trainingJobBuilder_ == null) { + return trainingJob_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.getDefaultInstance() : trainingJob_; + } else { + return trainingJobBuilder_.getMessage(); + } + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public Builder setTrainingJob(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob value) { + if (trainingJobBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + trainingJob_ = value; + onChanged(); + } else { + trainingJobBuilder_.setMessage(value); + } + + return this; + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public Builder setTrainingJob( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder builderForValue) { + if (trainingJobBuilder_ == null) { + trainingJob_ = builderForValue.build(); + onChanged(); + } else { + trainingJobBuilder_.setMessage(builderForValue.build()); + } + + return this; + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public Builder mergeTrainingJob(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob value) { + if (trainingJobBuilder_ == null) { + if (trainingJob_ != null) { + trainingJob_ = + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.newBuilder(trainingJob_).mergeFrom(value).buildPartial(); + } else { + trainingJob_ = value; + } + onChanged(); + } else { + trainingJobBuilder_.mergeFrom(value); + } + + return this; + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public Builder clearTrainingJob() { + if (trainingJobBuilder_ == null) { + trainingJob_ = null; + onChanged(); + } else { + trainingJob_ = null; + trainingJobBuilder_ = null; + } + + return this; + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder getTrainingJobBuilder() { + + onChanged(); + return getTrainingJobFieldBuilder().getBuilder(); + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder getTrainingJobOrBuilder() { + if (trainingJobBuilder_ != null) { + return trainingJobBuilder_.getMessageOrBuilder(); + } else { + return trainingJob_ == null ? + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.getDefaultInstance() : trainingJob_; + } + } + /** + *
+       * The underlying training job that the hyperparameter tuning job will launch during the process
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJob training_job = 1; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder> + getTrainingJobFieldBuilder() { + if (trainingJobBuilder_ == null) { + trainingJobBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder>( + getTrainingJob(), + getParentForChildren(), + isClean()); + trainingJob_ = null; + } + return trainingJobBuilder_; + } + + private long maxNumberOfTrainingJobs_ ; + /** + *
+       * The maximum number of training jobs that an hpo job can launch. For resource limit purpose.
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_number_of_training_jobs = 2; + */ + public long getMaxNumberOfTrainingJobs() { + return maxNumberOfTrainingJobs_; + } + /** + *
+       * The maximum number of training jobs that an hpo job can launch. For resource limit purpose.
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_number_of_training_jobs = 2; + */ + public Builder setMaxNumberOfTrainingJobs(long value) { + + maxNumberOfTrainingJobs_ = value; + onChanged(); + return this; + } + /** + *
+       * The maximum number of training jobs that an hpo job can launch. For resource limit purpose.
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_number_of_training_jobs = 2; + */ + public Builder clearMaxNumberOfTrainingJobs() { + + maxNumberOfTrainingJobs_ = 0L; + onChanged(); + return this; + } + + private long maxParallelTrainingJobs_ ; + /** + *
+       * The maximum number of concurrent training job that an hpo job can launch
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_parallel_training_jobs = 3; + */ + public long getMaxParallelTrainingJobs() { + return maxParallelTrainingJobs_; + } + /** + *
+       * The maximum number of concurrent training job that an hpo job can launch
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_parallel_training_jobs = 3; + */ + public Builder setMaxParallelTrainingJobs(long value) { + + maxParallelTrainingJobs_ = value; + onChanged(); + return this; + } + /** + *
+       * The maximum number of concurrent training job that an hpo job can launch
+       * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html
+       * 
+ * + * int64 max_parallel_training_jobs = 3; + */ + public Builder clearMaxParallelTrainingJobs() { + + maxParallelTrainingJobs_ = 0L; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterTuningJob parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterTuningJob(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJob getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface HyperparameterTuningObjectiveTypeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+   * with respect to the specified metric.
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType} + */ + public static final class HyperparameterTuningObjectiveType extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + HyperparameterTuningObjectiveTypeOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterTuningObjectiveType.newBuilder() to construct. + private HyperparameterTuningObjectiveType(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterTuningObjectiveType() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterTuningObjectiveType( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * MINIMIZE = 0; + */ + MINIMIZE(0), + /** + * MAXIMIZE = 1; + */ + MAXIMIZE(1), + UNRECOGNIZED(-1), + ; + + /** + * MINIMIZE = 0; + */ + public static final int MINIMIZE_VALUE = 0; + /** + * MAXIMIZE = 1; + */ + public static final int MAXIMIZE_VALUE = 1; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return MINIMIZE; + case 1: return MAXIMIZE; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+     * with respect to the specified metric.
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveTypeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterTuningObjectiveType parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterTuningObjectiveType(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface HyperparameterTuningObjectiveOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+     * with respect to the specified metric.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + int getObjectiveTypeValue(); + /** + *
+     * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+     * with respect to the specified metric.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value getObjectiveType(); + + /** + *
+     * The target metric name, which is the user-defined name of the metric specified in the
+     * training job's algorithm specification
+     * 
+ * + * string metric_name = 2; + */ + java.lang.String getMetricName(); + /** + *
+     * The target metric name, which is the user-defined name of the metric specified in the
+     * training job's algorithm specification
+     * 
+ * + * string metric_name = 2; + */ + com.google.protobuf.ByteString + getMetricNameBytes(); + } + /** + *
+   * The target metric and the objective of the hyperparameter tuning.
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningObjective} + */ + public static final class HyperparameterTuningObjective extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + HyperparameterTuningObjectiveOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterTuningObjective.newBuilder() to construct. + private HyperparameterTuningObjective(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterTuningObjective() { + objectiveType_ = 0; + metricName_ = ""; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterTuningObjective( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 8: { + int rawValue = input.readEnum(); + + objectiveType_ = rawValue; + break; + } + case 18: { + java.lang.String s = input.readStringRequireUtf8(); + + metricName_ = s; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder.class); + } + + public static final int OBJECTIVE_TYPE_FIELD_NUMBER = 1; + private int objectiveType_; + /** + *
+     * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+     * with respect to the specified metric.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public int getObjectiveTypeValue() { + return objectiveType_; + } + /** + *
+     * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+     * with respect to the specified metric.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value getObjectiveType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.valueOf(objectiveType_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.UNRECOGNIZED : result; + } + + public static final int METRIC_NAME_FIELD_NUMBER = 2; + private volatile java.lang.Object metricName_; + /** + *
+     * The target metric name, which is the user-defined name of the metric specified in the
+     * training job's algorithm specification
+     * 
+ * + * string metric_name = 2; + */ + public java.lang.String getMetricName() { + java.lang.Object ref = metricName_; + if (ref instanceof java.lang.String) { + return (java.lang.String) ref; + } else { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + metricName_ = s; + return s; + } + } + /** + *
+     * The target metric name, which is the user-defined name of the metric specified in the
+     * training job's algorithm specification
+     * 
+ * + * string metric_name = 2; + */ + public com.google.protobuf.ByteString + getMetricNameBytes() { + java.lang.Object ref = metricName_; + if (ref instanceof java.lang.String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + metricName_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (objectiveType_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.MINIMIZE.getNumber()) { + output.writeEnum(1, objectiveType_); + } + if (!getMetricNameBytes().isEmpty()) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 2, metricName_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (objectiveType_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.MINIMIZE.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(1, objectiveType_); + } + if (!getMetricNameBytes().isEmpty()) { + size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, metricName_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective) obj; + + if (objectiveType_ != other.objectiveType_) return false; + if (!getMetricName() + .equals(other.getMetricName())) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + OBJECTIVE_TYPE_FIELD_NUMBER; + hash = (53 * hash) + objectiveType_; + hash = (37 * hash) + METRIC_NAME_FIELD_NUMBER; + hash = (53 * hash) + getMetricName().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningObjective} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + objectiveType_ = 0; + + metricName_ = ""; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective(this); + result.objectiveType_ = objectiveType_; + result.metricName_ = metricName_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.getDefaultInstance()) return this; + if (other.objectiveType_ != 0) { + setObjectiveTypeValue(other.getObjectiveTypeValue()); + } + if (!other.getMetricName().isEmpty()) { + metricName_ = other.metricName_; + onChanged(); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private int objectiveType_ = 0; + /** + *
+       * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+       * with respect to the specified metric.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public int getObjectiveTypeValue() { + return objectiveType_; + } + /** + *
+       * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+       * with respect to the specified metric.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public Builder setObjectiveTypeValue(int value) { + objectiveType_ = value; + onChanged(); + return this; + } + /** + *
+       * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+       * with respect to the specified metric.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value getObjectiveType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.valueOf(objectiveType_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value.UNRECOGNIZED : result; + } + /** + *
+       * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+       * with respect to the specified metric.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public Builder setObjectiveType(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveType.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + objectiveType_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job
+       * with respect to the specified metric.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value objective_type = 1; + */ + public Builder clearObjectiveType() { + + objectiveType_ = 0; + onChanged(); + return this; + } + + private java.lang.Object metricName_ = ""; + /** + *
+       * The target metric name, which is the user-defined name of the metric specified in the
+       * training job's algorithm specification
+       * 
+ * + * string metric_name = 2; + */ + public java.lang.String getMetricName() { + java.lang.Object ref = metricName_; + if (!(ref instanceof java.lang.String)) { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + metricName_ = s; + return s; + } else { + return (java.lang.String) ref; + } + } + /** + *
+       * The target metric name, which is the user-defined name of the metric specified in the
+       * training job's algorithm specification
+       * 
+ * + * string metric_name = 2; + */ + public com.google.protobuf.ByteString + getMetricNameBytes() { + java.lang.Object ref = metricName_; + if (ref instanceof String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + metricName_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + /** + *
+       * The target metric name, which is the user-defined name of the metric specified in the
+       * training job's algorithm specification
+       * 
+ * + * string metric_name = 2; + */ + public Builder setMetricName( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + + metricName_ = value; + onChanged(); + return this; + } + /** + *
+       * The target metric name, which is the user-defined name of the metric specified in the
+       * training job's algorithm specification
+       * 
+ * + * string metric_name = 2; + */ + public Builder clearMetricName() { + + metricName_ = getDefaultInstance().getMetricName(); + onChanged(); + return this; + } + /** + *
+       * The target metric name, which is the user-defined name of the metric specified in the
+       * training job's algorithm specification
+       * 
+ * + * string metric_name = 2; + */ + public Builder setMetricNameBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + + metricName_ = value; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterTuningObjective parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterTuningObjective(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface HyperparameterTuningStrategyOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * Setting the strategy used when searching in the hyperparameter space
+   * Refer this doc for more details:
+   * https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningStrategy} + */ + public static final class HyperparameterTuningStrategy extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + HyperparameterTuningStrategyOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterTuningStrategy.newBuilder() to construct. + private HyperparameterTuningStrategy(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterTuningStrategy() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterTuningStrategy( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * BAYESIAN = 0; + */ + BAYESIAN(0), + /** + * RANDOM = 1; + */ + RANDOM(1), + UNRECOGNIZED(-1), + ; + + /** + * BAYESIAN = 0; + */ + public static final int BAYESIAN_VALUE = 0; + /** + * RANDOM = 1; + */ + public static final int RANDOM_VALUE = 1; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return BAYESIAN; + case 1: return RANDOM; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * Setting the strategy used when searching in the hyperparameter space
+     * Refer this doc for more details:
+     * https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningStrategy} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategyOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterTuningStrategy parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterTuningStrategy(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface TrainingJobEarlyStoppingTypeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+   * a hyperparameter tuning job can be stopping early.
+   * Note that there's only a subset of built-in algorithms that supports early stopping.
+   * see: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-early-stopping.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType} + */ + public static final class TrainingJobEarlyStoppingType extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + TrainingJobEarlyStoppingTypeOrBuilder { + private static final long serialVersionUID = 0L; + // Use TrainingJobEarlyStoppingType.newBuilder() to construct. + private TrainingJobEarlyStoppingType(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private TrainingJobEarlyStoppingType() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private TrainingJobEarlyStoppingType( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * OFF = 0; + */ + OFF(0), + /** + * AUTO = 1; + */ + AUTO(1), + UNRECOGNIZED(-1), + ; + + /** + * OFF = 0; + */ + public static final int OFF_VALUE = 0; + /** + * AUTO = 1; + */ + public static final int AUTO_VALUE = 1; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return OFF; + case 1: return AUTO; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+     * a hyperparameter tuning job can be stopping early.
+     * Note that there's only a subset of built-in algorithms that supports early stopping.
+     * see: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-early-stopping.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingTypeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public TrainingJobEarlyStoppingType parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new TrainingJobEarlyStoppingType(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface HyperparameterTuningJobConfigOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + boolean hasHyperparameterRanges(); + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getHyperparameterRanges(); + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder getHyperparameterRangesOrBuilder(); + + /** + *
+     * Setting the strategy used when searching in the hyperparameter space
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + int getTuningStrategyValue(); + /** + *
+     * Setting the strategy used when searching in the hyperparameter space
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value getTuningStrategy(); + + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + boolean hasTuningObjective(); + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getTuningObjective(); + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder getTuningObjectiveOrBuilder(); + + /** + *
+     * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+     * a hyperparameter tuning job can be stopping early.
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + int getTrainingJobEarlyStoppingTypeValue(); + /** + *
+     * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+     * a hyperparameter tuning job can be stopping early.
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value getTrainingJobEarlyStoppingType(); + } + /** + *
+   * The specification of the hyperparameter tuning process
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig} + */ + public static final class HyperparameterTuningJobConfig extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + HyperparameterTuningJobConfigOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterTuningJobConfig.newBuilder() to construct. + private HyperparameterTuningJobConfig(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterTuningJobConfig() { + tuningStrategy_ = 0; + trainingJobEarlyStoppingType_ = 0; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterTuningJobConfig( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder subBuilder = null; + if (hyperparameterRanges_ != null) { + subBuilder = hyperparameterRanges_.toBuilder(); + } + hyperparameterRanges_ = input.readMessage(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom(hyperparameterRanges_); + hyperparameterRanges_ = subBuilder.buildPartial(); + } + + break; + } + case 16: { + int rawValue = input.readEnum(); + + tuningStrategy_ = rawValue; + break; + } + case 26: { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder subBuilder = null; + if (tuningObjective_ != null) { + subBuilder = tuningObjective_.toBuilder(); + } + tuningObjective_ = input.readMessage(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom(tuningObjective_); + tuningObjective_ = subBuilder.buildPartial(); + } + + break; + } + case 32: { + int rawValue = input.readEnum(); + + trainingJobEarlyStoppingType_ = rawValue; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.Builder.class); + } + + public static final int HYPERPARAMETER_RANGES_FIELD_NUMBER = 1; + private flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges hyperparameterRanges_; + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public boolean hasHyperparameterRanges() { + return hyperparameterRanges_ != null; + } + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getHyperparameterRanges() { + return hyperparameterRanges_ == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.getDefaultInstance() : hyperparameterRanges_; + } + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder getHyperparameterRangesOrBuilder() { + return getHyperparameterRanges(); + } + + public static final int TUNING_STRATEGY_FIELD_NUMBER = 2; + private int tuningStrategy_; + /** + *
+     * Setting the strategy used when searching in the hyperparameter space
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public int getTuningStrategyValue() { + return tuningStrategy_; + } + /** + *
+     * Setting the strategy used when searching in the hyperparameter space
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value getTuningStrategy() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.valueOf(tuningStrategy_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.UNRECOGNIZED : result; + } + + public static final int TUNING_OBJECTIVE_FIELD_NUMBER = 3; + private flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective tuningObjective_; + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public boolean hasTuningObjective() { + return tuningObjective_ != null; + } + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getTuningObjective() { + return tuningObjective_ == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.getDefaultInstance() : tuningObjective_; + } + /** + *
+     * The target metric and the objective of the hyperparameter tuning.
+     * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder getTuningObjectiveOrBuilder() { + return getTuningObjective(); + } + + public static final int TRAINING_JOB_EARLY_STOPPING_TYPE_FIELD_NUMBER = 4; + private int trainingJobEarlyStoppingType_; + /** + *
+     * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+     * a hyperparameter tuning job can be stopping early.
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public int getTrainingJobEarlyStoppingTypeValue() { + return trainingJobEarlyStoppingType_; + } + /** + *
+     * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+     * a hyperparameter tuning job can be stopping early.
+     * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value getTrainingJobEarlyStoppingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.valueOf(trainingJobEarlyStoppingType_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.UNRECOGNIZED : result; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (hyperparameterRanges_ != null) { + output.writeMessage(1, getHyperparameterRanges()); + } + if (tuningStrategy_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.BAYESIAN.getNumber()) { + output.writeEnum(2, tuningStrategy_); + } + if (tuningObjective_ != null) { + output.writeMessage(3, getTuningObjective()); + } + if (trainingJobEarlyStoppingType_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.OFF.getNumber()) { + output.writeEnum(4, trainingJobEarlyStoppingType_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (hyperparameterRanges_ != null) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(1, getHyperparameterRanges()); + } + if (tuningStrategy_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.BAYESIAN.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(2, tuningStrategy_); + } + if (tuningObjective_ != null) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(3, getTuningObjective()); + } + if (trainingJobEarlyStoppingType_ != flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.OFF.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(4, trainingJobEarlyStoppingType_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig other = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig) obj; + + if (hasHyperparameterRanges() != other.hasHyperparameterRanges()) return false; + if (hasHyperparameterRanges()) { + if (!getHyperparameterRanges() + .equals(other.getHyperparameterRanges())) return false; + } + if (tuningStrategy_ != other.tuningStrategy_) return false; + if (hasTuningObjective() != other.hasTuningObjective()) return false; + if (hasTuningObjective()) { + if (!getTuningObjective() + .equals(other.getTuningObjective())) return false; + } + if (trainingJobEarlyStoppingType_ != other.trainingJobEarlyStoppingType_) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + if (hasHyperparameterRanges()) { + hash = (37 * hash) + HYPERPARAMETER_RANGES_FIELD_NUMBER; + hash = (53 * hash) + getHyperparameterRanges().hashCode(); + } + hash = (37 * hash) + TUNING_STRATEGY_FIELD_NUMBER; + hash = (53 * hash) + tuningStrategy_; + if (hasTuningObjective()) { + hash = (37 * hash) + TUNING_OBJECTIVE_FIELD_NUMBER; + hash = (53 * hash) + getTuningObjective().hashCode(); + } + hash = (37 * hash) + TRAINING_JOB_EARLY_STOPPING_TYPE_FIELD_NUMBER; + hash = (53 * hash) + trainingJobEarlyStoppingType_; + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * The specification of the hyperparameter tuning process
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfigOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.class, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + if (hyperparameterRangesBuilder_ == null) { + hyperparameterRanges_ = null; + } else { + hyperparameterRanges_ = null; + hyperparameterRangesBuilder_ = null; + } + tuningStrategy_ = 0; + + if (tuningObjectiveBuilder_ == null) { + tuningObjective_ = null; + } else { + tuningObjective_ = null; + tuningObjectiveBuilder_ = null; + } + trainingJobEarlyStoppingType_ = 0; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig build() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig buildPartial() { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig result = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig(this); + if (hyperparameterRangesBuilder_ == null) { + result.hyperparameterRanges_ = hyperparameterRanges_; + } else { + result.hyperparameterRanges_ = hyperparameterRangesBuilder_.build(); + } + result.tuningStrategy_ = tuningStrategy_; + if (tuningObjectiveBuilder_ == null) { + result.tuningObjective_ = tuningObjective_; + } else { + result.tuningObjective_ = tuningObjectiveBuilder_.build(); + } + result.trainingJobEarlyStoppingType_ = trainingJobEarlyStoppingType_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig) { + return mergeFrom((flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig other) { + if (other == flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig.getDefaultInstance()) return this; + if (other.hasHyperparameterRanges()) { + mergeHyperparameterRanges(other.getHyperparameterRanges()); + } + if (other.tuningStrategy_ != 0) { + setTuningStrategyValue(other.getTuningStrategyValue()); + } + if (other.hasTuningObjective()) { + mergeTuningObjective(other.getTuningObjective()); + } + if (other.trainingJobEarlyStoppingType_ != 0) { + setTrainingJobEarlyStoppingTypeValue(other.getTrainingJobEarlyStoppingTypeValue()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges hyperparameterRanges_; + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder> hyperparameterRangesBuilder_; + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public boolean hasHyperparameterRanges() { + return hyperparameterRangesBuilder_ != null || hyperparameterRanges_ != null; + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getHyperparameterRanges() { + if (hyperparameterRangesBuilder_ == null) { + return hyperparameterRanges_ == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.getDefaultInstance() : hyperparameterRanges_; + } else { + return hyperparameterRangesBuilder_.getMessage(); + } + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public Builder setHyperparameterRanges(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges value) { + if (hyperparameterRangesBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + hyperparameterRanges_ = value; + onChanged(); + } else { + hyperparameterRangesBuilder_.setMessage(value); + } + + return this; + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public Builder setHyperparameterRanges( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder builderForValue) { + if (hyperparameterRangesBuilder_ == null) { + hyperparameterRanges_ = builderForValue.build(); + onChanged(); + } else { + hyperparameterRangesBuilder_.setMessage(builderForValue.build()); + } + + return this; + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public Builder mergeHyperparameterRanges(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges value) { + if (hyperparameterRangesBuilder_ == null) { + if (hyperparameterRanges_ != null) { + hyperparameterRanges_ = + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.newBuilder(hyperparameterRanges_).mergeFrom(value).buildPartial(); + } else { + hyperparameterRanges_ = value; + } + onChanged(); + } else { + hyperparameterRangesBuilder_.mergeFrom(value); + } + + return this; + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public Builder clearHyperparameterRanges() { + if (hyperparameterRangesBuilder_ == null) { + hyperparameterRanges_ = null; + onChanged(); + } else { + hyperparameterRanges_ = null; + hyperparameterRangesBuilder_ = null; + } + + return this; + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder getHyperparameterRangesBuilder() { + + onChanged(); + return getHyperparameterRangesFieldBuilder().getBuilder(); + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder getHyperparameterRangesOrBuilder() { + if (hyperparameterRangesBuilder_ != null) { + return hyperparameterRangesBuilder_.getMessageOrBuilder(); + } else { + return hyperparameterRanges_ == null ? + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.getDefaultInstance() : hyperparameterRanges_; + } + } + /** + *
+       * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+       * 
+ * + * .flyteidl.plugins.sagemaker.ParameterRanges hyperparameter_ranges = 1; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder> + getHyperparameterRangesFieldBuilder() { + if (hyperparameterRangesBuilder_ == null) { + hyperparameterRangesBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder>( + getHyperparameterRanges(), + getParentForChildren(), + isClean()); + hyperparameterRanges_ = null; + } + return hyperparameterRangesBuilder_; + } + + private int tuningStrategy_ = 0; + /** + *
+       * Setting the strategy used when searching in the hyperparameter space
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public int getTuningStrategyValue() { + return tuningStrategy_; + } + /** + *
+       * Setting the strategy used when searching in the hyperparameter space
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public Builder setTuningStrategyValue(int value) { + tuningStrategy_ = value; + onChanged(); + return this; + } + /** + *
+       * Setting the strategy used when searching in the hyperparameter space
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value getTuningStrategy() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.valueOf(tuningStrategy_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value.UNRECOGNIZED : result; + } + /** + *
+       * Setting the strategy used when searching in the hyperparameter space
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public Builder setTuningStrategy(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningStrategy.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + tuningStrategy_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * Setting the strategy used when searching in the hyperparameter space
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value tuning_strategy = 2; + */ + public Builder clearTuningStrategy() { + + tuningStrategy_ = 0; + onChanged(); + return this; + } + + private flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective tuningObjective_; + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder> tuningObjectiveBuilder_; + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public boolean hasTuningObjective() { + return tuningObjectiveBuilder_ != null || tuningObjective_ != null; + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective getTuningObjective() { + if (tuningObjectiveBuilder_ == null) { + return tuningObjective_ == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.getDefaultInstance() : tuningObjective_; + } else { + return tuningObjectiveBuilder_.getMessage(); + } + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public Builder setTuningObjective(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective value) { + if (tuningObjectiveBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + tuningObjective_ = value; + onChanged(); + } else { + tuningObjectiveBuilder_.setMessage(value); + } + + return this; + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public Builder setTuningObjective( + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder builderForValue) { + if (tuningObjectiveBuilder_ == null) { + tuningObjective_ = builderForValue.build(); + onChanged(); + } else { + tuningObjectiveBuilder_.setMessage(builderForValue.build()); + } + + return this; + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public Builder mergeTuningObjective(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective value) { + if (tuningObjectiveBuilder_ == null) { + if (tuningObjective_ != null) { + tuningObjective_ = + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.newBuilder(tuningObjective_).mergeFrom(value).buildPartial(); + } else { + tuningObjective_ = value; + } + onChanged(); + } else { + tuningObjectiveBuilder_.mergeFrom(value); + } + + return this; + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public Builder clearTuningObjective() { + if (tuningObjectiveBuilder_ == null) { + tuningObjective_ = null; + onChanged(); + } else { + tuningObjective_ = null; + tuningObjectiveBuilder_ = null; + } + + return this; + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder getTuningObjectiveBuilder() { + + onChanged(); + return getTuningObjectiveFieldBuilder().getBuilder(); + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder getTuningObjectiveOrBuilder() { + if (tuningObjectiveBuilder_ != null) { + return tuningObjectiveBuilder_.getMessageOrBuilder(); + } else { + return tuningObjective_ == null ? + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.getDefaultInstance() : tuningObjective_; + } + } + /** + *
+       * The target metric and the objective of the hyperparameter tuning.
+       * 
+ * + * .flyteidl.plugins.sagemaker.HyperparameterTuningObjective tuning_objective = 3; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder> + getTuningObjectiveFieldBuilder() { + if (tuningObjectiveBuilder_ == null) { + tuningObjectiveBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjective.Builder, flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningObjectiveOrBuilder>( + getTuningObjective(), + getParentForChildren(), + isClean()); + tuningObjective_ = null; + } + return tuningObjectiveBuilder_; + } + + private int trainingJobEarlyStoppingType_ = 0; + /** + *
+       * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+       * a hyperparameter tuning job can be stopping early.
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public int getTrainingJobEarlyStoppingTypeValue() { + return trainingJobEarlyStoppingType_; + } + /** + *
+       * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+       * a hyperparameter tuning job can be stopping early.
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public Builder setTrainingJobEarlyStoppingTypeValue(int value) { + trainingJobEarlyStoppingType_ = value; + onChanged(); + return this; + } + /** + *
+       * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+       * a hyperparameter tuning job can be stopping early.
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value getTrainingJobEarlyStoppingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value result = flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.valueOf(trainingJobEarlyStoppingType_); + return result == null ? flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value.UNRECOGNIZED : result; + } + /** + *
+       * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+       * a hyperparameter tuning job can be stopping early.
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public Builder setTrainingJobEarlyStoppingType(flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.TrainingJobEarlyStoppingType.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + trainingJobEarlyStoppingType_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * When the training jobs launched by the hyperparameter tuning job are not improving significantly,
+       * a hyperparameter tuning job can be stopping early.
+       * 
+ * + * .flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; + */ + public Builder clearTrainingJobEarlyStoppingType() { + + trainingJobEarlyStoppingType_ = 0; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + private static final flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig(); + } + + public static flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterTuningJobConfig parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterTuningJobConfig(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.HyperparameterTuningJobOuterClass.HyperparameterTuningJobConfig getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_fieldAccessorTable; + + public static com.google.protobuf.Descriptors.FileDescriptor + getDescriptor() { + return descriptor; + } + private static com.google.protobuf.Descriptors.FileDescriptor + descriptor; + static { + java.lang.String[] descriptorData = { + "\n:flyteidl/plugins/sagemaker/hyperparame" + + "ter_tuning_job.proto\022\032flyteidl.plugins.s" + + "agemaker\0321flyteidl/plugins/sagemaker/par" + + "ameter_ranges.proto\032-flyteidl/plugins/sa" + + "gemaker/training_job.proto\"\241\001\n\027Hyperpara" + + "meterTuningJob\022=\n\014training_job\030\001 \001(\0132\'.f" + + "lyteidl.plugins.sagemaker.TrainingJob\022#\n" + + "\033max_number_of_training_jobs\030\002 \001(\003\022\"\n\032ma" + + "x_parallel_training_jobs\030\003 \001(\003\"H\n!Hyperp" + + "arameterTuningObjectiveType\"#\n\005Value\022\014\n\010" + + "MINIMIZE\020\000\022\014\n\010MAXIMIZE\020\001\"\221\001\n\035Hyperparame" + + "terTuningObjective\022[\n\016objective_type\030\001 \001" + + "(\0162C.flyteidl.plugins.sagemaker.Hyperpar" + + "ameterTuningObjectiveType.Value\022\023\n\013metri" + + "c_name\030\002 \001(\t\"A\n\034HyperparameterTuningStra" + + "tegy\"!\n\005Value\022\014\n\010BAYESIAN\020\000\022\n\n\006RANDOM\020\001\"" + + ":\n\034TrainingJobEarlyStoppingType\"\032\n\005Value" + + "\022\007\n\003OFF\020\000\022\010\n\004AUTO\020\001\"\203\003\n\035HyperparameterTu" + + "ningJobConfig\022J\n\025hyperparameter_ranges\030\001" + + " \001(\0132+.flyteidl.plugins.sagemaker.Parame" + + "terRanges\022W\n\017tuning_strategy\030\002 \001(\0162>.fly" + + "teidl.plugins.sagemaker.HyperparameterTu" + + "ningStrategy.Value\022S\n\020tuning_objective\030\003" + + " \001(\01329.flyteidl.plugins.sagemaker.Hyperp" + + "arameterTuningObjective\022h\n training_job_" + + "early_stopping_type\030\004 \001(\0162>.flyteidl.plu" + + "gins.sagemaker.TrainingJobEarlyStoppingT" + + "ype.ValueB5Z3github.com/lyft/flyteidl/ge" + + "n/pb-go/flyteidl/pluginsb\006proto3" + }; + com.google.protobuf.Descriptors.FileDescriptor.InternalDescriptorAssigner assigner = + new com.google.protobuf.Descriptors.FileDescriptor. InternalDescriptorAssigner() { + public com.google.protobuf.ExtensionRegistry assignDescriptors( + com.google.protobuf.Descriptors.FileDescriptor root) { + descriptor = root; + return null; + } + }; + com.google.protobuf.Descriptors.FileDescriptor + .internalBuildGeneratedFileFrom(descriptorData, + new com.google.protobuf.Descriptors.FileDescriptor[] { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.getDescriptor(), + flyteidl.plugins.sagemaker.TrainingJobOuterClass.getDescriptor(), + }, assigner); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor = + getDescriptor().getMessageTypes().get(0); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJob_descriptor, + new java.lang.String[] { "TrainingJob", "MaxNumberOfTrainingJobs", "MaxParallelTrainingJobs", }); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor = + getDescriptor().getMessageTypes().get(1); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjectiveType_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor = + getDescriptor().getMessageTypes().get(2); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningObjective_descriptor, + new java.lang.String[] { "ObjectiveType", "MetricName", }); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor = + getDescriptor().getMessageTypes().get(3); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningStrategy_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor = + getDescriptor().getMessageTypes().get(4); + internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_TrainingJobEarlyStoppingType_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor = + getDescriptor().getMessageTypes().get(5); + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterTuningJobConfig_descriptor, + new java.lang.String[] { "HyperparameterRanges", "TuningStrategy", "TuningObjective", "TrainingJobEarlyStoppingType", }); + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.getDescriptor(); + flyteidl.plugins.sagemaker.TrainingJobOuterClass.getDescriptor(); + } + + // @@protoc_insertion_point(outer_class_scope) +} diff --git a/gen/pb-java/flyteidl/plugins/sagemaker/ParameterRangesOuterClass.java b/gen/pb-java/flyteidl/plugins/sagemaker/ParameterRangesOuterClass.java new file mode 100644 index 000000000..d0ba674bd --- /dev/null +++ b/gen/pb-java/flyteidl/plugins/sagemaker/ParameterRangesOuterClass.java @@ -0,0 +1,4468 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +package flyteidl.plugins.sagemaker; + +public final class ParameterRangesOuterClass { + private ParameterRangesOuterClass() {} + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistryLite registry) { + } + + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistry registry) { + registerAllExtensions( + (com.google.protobuf.ExtensionRegistryLite) registry); + } + public interface HyperparameterScalingTypeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.HyperparameterScalingType) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * HyperparameterScalingType defines the way to increase or decrease the value of the hyperparameter
+   * For details, refer to: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * See examples of these scaling type, refer to: https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterScalingType} + */ + public static final class HyperparameterScalingType extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.HyperparameterScalingType) + HyperparameterScalingTypeOrBuilder { + private static final long serialVersionUID = 0L; + // Use HyperparameterScalingType.newBuilder() to construct. + private HyperparameterScalingType(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private HyperparameterScalingType() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private HyperparameterScalingType( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.HyperparameterScalingType.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * AUTO = 0; + */ + AUTO(0), + /** + * LINEAR = 1; + */ + LINEAR(1), + /** + * LOGARITHMIC = 2; + */ + LOGARITHMIC(2), + /** + * REVERSELOGARITHMIC = 3; + */ + REVERSELOGARITHMIC(3), + UNRECOGNIZED(-1), + ; + + /** + * AUTO = 0; + */ + public static final int AUTO_VALUE = 0; + /** + * LINEAR = 1; + */ + public static final int LINEAR_VALUE = 1; + /** + * LOGARITHMIC = 2; + */ + public static final int LOGARITHMIC_VALUE = 2; + /** + * REVERSELOGARITHMIC = 3; + */ + public static final int REVERSELOGARITHMIC_VALUE = 3; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return AUTO; + case 1: return LINEAR; + case 2: return LOGARITHMIC; + case 3: return REVERSELOGARITHMIC; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.HyperparameterScalingType.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * HyperparameterScalingType defines the way to increase or decrease the value of the hyperparameter
+     * For details, refer to: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * See examples of these scaling type, refer to: https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.HyperparameterScalingType} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.HyperparameterScalingType) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingTypeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.HyperparameterScalingType) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterScalingType) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public HyperparameterScalingType parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new HyperparameterScalingType(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface ContinuousParameterRangeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.ContinuousParameterRange) + com.google.protobuf.MessageOrBuilder { + + /** + * double max_value = 1; + */ + double getMaxValue(); + + /** + * double min_value = 2; + */ + double getMinValue(); + + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + int getScalingTypeValue(); + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType(); + } + /** + *
+   * ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing
+   * users to specify the search space of a floating-point hyperparameter
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ContinuousParameterRange} + */ + public static final class ContinuousParameterRange extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.ContinuousParameterRange) + ContinuousParameterRangeOrBuilder { + private static final long serialVersionUID = 0L; + // Use ContinuousParameterRange.newBuilder() to construct. + private ContinuousParameterRange(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private ContinuousParameterRange() { + scalingType_ = 0; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private ContinuousParameterRange( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 9: { + + maxValue_ = input.readDouble(); + break; + } + case 17: { + + minValue_ = input.readDouble(); + break; + } + case 24: { + int rawValue = input.readEnum(); + + scalingType_ = rawValue; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder.class); + } + + public static final int MAX_VALUE_FIELD_NUMBER = 1; + private double maxValue_; + /** + * double max_value = 1; + */ + public double getMaxValue() { + return maxValue_; + } + + public static final int MIN_VALUE_FIELD_NUMBER = 2; + private double minValue_; + /** + * double min_value = 2; + */ + public double getMinValue() { + return minValue_; + } + + public static final int SCALING_TYPE_FIELD_NUMBER = 3; + private int scalingType_; + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public int getScalingTypeValue() { + return scalingType_; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value result = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.valueOf(scalingType_); + return result == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.UNRECOGNIZED : result; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (maxValue_ != 0D) { + output.writeDouble(1, maxValue_); + } + if (minValue_ != 0D) { + output.writeDouble(2, minValue_); + } + if (scalingType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.AUTO.getNumber()) { + output.writeEnum(3, scalingType_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (maxValue_ != 0D) { + size += com.google.protobuf.CodedOutputStream + .computeDoubleSize(1, maxValue_); + } + if (minValue_ != 0D) { + size += com.google.protobuf.CodedOutputStream + .computeDoubleSize(2, minValue_); + } + if (scalingType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.AUTO.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(3, scalingType_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) obj; + + if (java.lang.Double.doubleToLongBits(getMaxValue()) + != java.lang.Double.doubleToLongBits( + other.getMaxValue())) return false; + if (java.lang.Double.doubleToLongBits(getMinValue()) + != java.lang.Double.doubleToLongBits( + other.getMinValue())) return false; + if (scalingType_ != other.scalingType_) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + MAX_VALUE_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + java.lang.Double.doubleToLongBits(getMaxValue())); + hash = (37 * hash) + MIN_VALUE_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + java.lang.Double.doubleToLongBits(getMinValue())); + hash = (37 * hash) + SCALING_TYPE_FIELD_NUMBER; + hash = (53 * hash) + scalingType_; + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing
+     * users to specify the search space of a floating-point hyperparameter
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ContinuousParameterRange} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.ContinuousParameterRange) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + maxValue_ = 0D; + + minValue_ = 0D; + + scalingType_ = 0; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange(this); + result.maxValue_ = maxValue_; + result.minValue_ = minValue_; + result.scalingType_ = scalingType_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance()) return this; + if (other.getMaxValue() != 0D) { + setMaxValue(other.getMaxValue()); + } + if (other.getMinValue() != 0D) { + setMinValue(other.getMinValue()); + } + if (other.scalingType_ != 0) { + setScalingTypeValue(other.getScalingTypeValue()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private double maxValue_ ; + /** + * double max_value = 1; + */ + public double getMaxValue() { + return maxValue_; + } + /** + * double max_value = 1; + */ + public Builder setMaxValue(double value) { + + maxValue_ = value; + onChanged(); + return this; + } + /** + * double max_value = 1; + */ + public Builder clearMaxValue() { + + maxValue_ = 0D; + onChanged(); + return this; + } + + private double minValue_ ; + /** + * double min_value = 2; + */ + public double getMinValue() { + return minValue_; + } + /** + * double min_value = 2; + */ + public Builder setMinValue(double value) { + + minValue_ = value; + onChanged(); + return this; + } + /** + * double min_value = 2; + */ + public Builder clearMinValue() { + + minValue_ = 0D; + onChanged(); + return this; + } + + private int scalingType_ = 0; + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public int getScalingTypeValue() { + return scalingType_; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder setScalingTypeValue(int value) { + scalingType_ = value; + onChanged(); + return this; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value result = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.valueOf(scalingType_); + return result == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.UNRECOGNIZED : result; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder setScalingType(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + scalingType_ = value.getNumber(); + onChanged(); + return this; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder clearScalingType() { + + scalingType_ = 0; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.ContinuousParameterRange) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ContinuousParameterRange) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public ContinuousParameterRange parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new ContinuousParameterRange(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface IntegerParameterRangeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.IntegerParameterRange) + com.google.protobuf.MessageOrBuilder { + + /** + * int64 max_value = 1; + */ + long getMaxValue(); + + /** + * int64 min_value = 2; + */ + long getMinValue(); + + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + int getScalingTypeValue(); + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType(); + } + /** + *
+   * IntegerParameterRange refers to a discrete range of hyperparameter values, allowing
+   * users to specify the search space of an integer hyperparameter
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.IntegerParameterRange} + */ + public static final class IntegerParameterRange extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.IntegerParameterRange) + IntegerParameterRangeOrBuilder { + private static final long serialVersionUID = 0L; + // Use IntegerParameterRange.newBuilder() to construct. + private IntegerParameterRange(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private IntegerParameterRange() { + scalingType_ = 0; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private IntegerParameterRange( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 8: { + + maxValue_ = input.readInt64(); + break; + } + case 16: { + + minValue_ = input.readInt64(); + break; + } + case 24: { + int rawValue = input.readEnum(); + + scalingType_ = rawValue; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder.class); + } + + public static final int MAX_VALUE_FIELD_NUMBER = 1; + private long maxValue_; + /** + * int64 max_value = 1; + */ + public long getMaxValue() { + return maxValue_; + } + + public static final int MIN_VALUE_FIELD_NUMBER = 2; + private long minValue_; + /** + * int64 min_value = 2; + */ + public long getMinValue() { + return minValue_; + } + + public static final int SCALING_TYPE_FIELD_NUMBER = 3; + private int scalingType_; + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public int getScalingTypeValue() { + return scalingType_; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value result = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.valueOf(scalingType_); + return result == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.UNRECOGNIZED : result; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (maxValue_ != 0L) { + output.writeInt64(1, maxValue_); + } + if (minValue_ != 0L) { + output.writeInt64(2, minValue_); + } + if (scalingType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.AUTO.getNumber()) { + output.writeEnum(3, scalingType_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (maxValue_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(1, maxValue_); + } + if (minValue_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(2, minValue_); + } + if (scalingType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.AUTO.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(3, scalingType_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) obj; + + if (getMaxValue() + != other.getMaxValue()) return false; + if (getMinValue() + != other.getMinValue()) return false; + if (scalingType_ != other.scalingType_) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + MAX_VALUE_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getMaxValue()); + hash = (37 * hash) + MIN_VALUE_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getMinValue()); + hash = (37 * hash) + SCALING_TYPE_FIELD_NUMBER; + hash = (53 * hash) + scalingType_; + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * IntegerParameterRange refers to a discrete range of hyperparameter values, allowing
+     * users to specify the search space of an integer hyperparameter
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.IntegerParameterRange} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.IntegerParameterRange) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + maxValue_ = 0L; + + minValue_ = 0L; + + scalingType_ = 0; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange(this); + result.maxValue_ = maxValue_; + result.minValue_ = minValue_; + result.scalingType_ = scalingType_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance()) return this; + if (other.getMaxValue() != 0L) { + setMaxValue(other.getMaxValue()); + } + if (other.getMinValue() != 0L) { + setMinValue(other.getMinValue()); + } + if (other.scalingType_ != 0) { + setScalingTypeValue(other.getScalingTypeValue()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private long maxValue_ ; + /** + * int64 max_value = 1; + */ + public long getMaxValue() { + return maxValue_; + } + /** + * int64 max_value = 1; + */ + public Builder setMaxValue(long value) { + + maxValue_ = value; + onChanged(); + return this; + } + /** + * int64 max_value = 1; + */ + public Builder clearMaxValue() { + + maxValue_ = 0L; + onChanged(); + return this; + } + + private long minValue_ ; + /** + * int64 min_value = 2; + */ + public long getMinValue() { + return minValue_; + } + /** + * int64 min_value = 2; + */ + public Builder setMinValue(long value) { + + minValue_ = value; + onChanged(); + return this; + } + /** + * int64 min_value = 2; + */ + public Builder clearMinValue() { + + minValue_ = 0L; + onChanged(); + return this; + } + + private int scalingType_ = 0; + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public int getScalingTypeValue() { + return scalingType_; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder setScalingTypeValue(int value) { + scalingType_ = value; + onChanged(); + return this; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value getScalingType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value result = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.valueOf(scalingType_); + return result == null ? flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value.UNRECOGNIZED : result; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder setScalingType(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.HyperparameterScalingType.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + scalingType_ = value.getNumber(); + onChanged(); + return this; + } + /** + * .flyteidl.plugins.sagemaker.HyperparameterScalingType.Value scaling_type = 3; + */ + public Builder clearScalingType() { + + scalingType_ = 0; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.IntegerParameterRange) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.IntegerParameterRange) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public IntegerParameterRange parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new IntegerParameterRange(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface CategoricalParameterRangeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.CategoricalParameterRange) + com.google.protobuf.MessageOrBuilder { + + /** + * repeated string values = 1; + */ + java.util.List + getValuesList(); + /** + * repeated string values = 1; + */ + int getValuesCount(); + /** + * repeated string values = 1; + */ + java.lang.String getValues(int index); + /** + * repeated string values = 1; + */ + com.google.protobuf.ByteString + getValuesBytes(int index); + } + /** + *
+   * ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing
+   * users to specify the search space of a floating-point hyperparameter
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.CategoricalParameterRange} + */ + public static final class CategoricalParameterRange extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.CategoricalParameterRange) + CategoricalParameterRangeOrBuilder { + private static final long serialVersionUID = 0L; + // Use CategoricalParameterRange.newBuilder() to construct. + private CategoricalParameterRange(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private CategoricalParameterRange() { + values_ = com.google.protobuf.LazyStringArrayList.EMPTY; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private CategoricalParameterRange( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + java.lang.String s = input.readStringRequireUtf8(); + if (!((mutable_bitField0_ & 0x00000001) != 0)) { + values_ = new com.google.protobuf.LazyStringArrayList(); + mutable_bitField0_ |= 0x00000001; + } + values_.add(s); + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + if (((mutable_bitField0_ & 0x00000001) != 0)) { + values_ = values_.getUnmodifiableView(); + } + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder.class); + } + + public static final int VALUES_FIELD_NUMBER = 1; + private com.google.protobuf.LazyStringList values_; + /** + * repeated string values = 1; + */ + public com.google.protobuf.ProtocolStringList + getValuesList() { + return values_; + } + /** + * repeated string values = 1; + */ + public int getValuesCount() { + return values_.size(); + } + /** + * repeated string values = 1; + */ + public java.lang.String getValues(int index) { + return values_.get(index); + } + /** + * repeated string values = 1; + */ + public com.google.protobuf.ByteString + getValuesBytes(int index) { + return values_.getByteString(index); + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + for (int i = 0; i < values_.size(); i++) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 1, values_.getRaw(i)); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + { + int dataSize = 0; + for (int i = 0; i < values_.size(); i++) { + dataSize += computeStringSizeNoTag(values_.getRaw(i)); + } + size += dataSize; + size += 1 * getValuesList().size(); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) obj; + + if (!getValuesList() + .equals(other.getValuesList())) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + if (getValuesCount() > 0) { + hash = (37 * hash) + VALUES_FIELD_NUMBER; + hash = (53 * hash) + getValuesList().hashCode(); + } + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing
+     * users to specify the search space of a floating-point hyperparameter
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.CategoricalParameterRange} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.CategoricalParameterRange) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + values_ = com.google.protobuf.LazyStringArrayList.EMPTY; + bitField0_ = (bitField0_ & ~0x00000001); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange(this); + int from_bitField0_ = bitField0_; + if (((bitField0_ & 0x00000001) != 0)) { + values_ = values_.getUnmodifiableView(); + bitField0_ = (bitField0_ & ~0x00000001); + } + result.values_ = values_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance()) return this; + if (!other.values_.isEmpty()) { + if (values_.isEmpty()) { + values_ = other.values_; + bitField0_ = (bitField0_ & ~0x00000001); + } else { + ensureValuesIsMutable(); + values_.addAll(other.values_); + } + onChanged(); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + private int bitField0_; + + private com.google.protobuf.LazyStringList values_ = com.google.protobuf.LazyStringArrayList.EMPTY; + private void ensureValuesIsMutable() { + if (!((bitField0_ & 0x00000001) != 0)) { + values_ = new com.google.protobuf.LazyStringArrayList(values_); + bitField0_ |= 0x00000001; + } + } + /** + * repeated string values = 1; + */ + public com.google.protobuf.ProtocolStringList + getValuesList() { + return values_.getUnmodifiableView(); + } + /** + * repeated string values = 1; + */ + public int getValuesCount() { + return values_.size(); + } + /** + * repeated string values = 1; + */ + public java.lang.String getValues(int index) { + return values_.get(index); + } + /** + * repeated string values = 1; + */ + public com.google.protobuf.ByteString + getValuesBytes(int index) { + return values_.getByteString(index); + } + /** + * repeated string values = 1; + */ + public Builder setValues( + int index, java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + ensureValuesIsMutable(); + values_.set(index, value); + onChanged(); + return this; + } + /** + * repeated string values = 1; + */ + public Builder addValues( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + ensureValuesIsMutable(); + values_.add(value); + onChanged(); + return this; + } + /** + * repeated string values = 1; + */ + public Builder addAllValues( + java.lang.Iterable values) { + ensureValuesIsMutable(); + com.google.protobuf.AbstractMessageLite.Builder.addAll( + values, values_); + onChanged(); + return this; + } + /** + * repeated string values = 1; + */ + public Builder clearValues() { + values_ = com.google.protobuf.LazyStringArrayList.EMPTY; + bitField0_ = (bitField0_ & ~0x00000001); + onChanged(); + return this; + } + /** + * repeated string values = 1; + */ + public Builder addValuesBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + ensureValuesIsMutable(); + values_.add(value); + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.CategoricalParameterRange) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.CategoricalParameterRange) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public CategoricalParameterRange parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new CategoricalParameterRange(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface ParameterRangeOneOfOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + com.google.protobuf.MessageOrBuilder { + + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + boolean hasContinuousParameterRange(); + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getContinuousParameterRange(); + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder getContinuousParameterRangeOrBuilder(); + + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + boolean hasIntegerParameterRange(); + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getIntegerParameterRange(); + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder getIntegerParameterRangeOrBuilder(); + + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + boolean hasCategoricalParameterRange(); + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getCategoricalParameterRange(); + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder getCategoricalParameterRangeOrBuilder(); + + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.ParameterRangeTypeCase getParameterRangeTypeCase(); + } + /** + *
+   * ParameterRangeOneOf describes a single ParameterRange, which is a one-of structure that can be one of
+   * the three possible types: ContinuousParameterRange, IntegerParameterRange, and CategoricalParameterRange.
+   * This one-of structure in Flyte enables specifying a Parameter in a type-safe manner
+   * See: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ParameterRangeOneOf} + */ + public static final class ParameterRangeOneOf extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + ParameterRangeOneOfOrBuilder { + private static final long serialVersionUID = 0L; + // Use ParameterRangeOneOf.newBuilder() to construct. + private ParameterRangeOneOf(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private ParameterRangeOneOf() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private ParameterRangeOneOf( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder subBuilder = null; + if (parameterRangeTypeCase_ == 1) { + subBuilder = ((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_).toBuilder(); + } + parameterRangeType_ = + input.readMessage(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_); + parameterRangeType_ = subBuilder.buildPartial(); + } + parameterRangeTypeCase_ = 1; + break; + } + case 18: { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder subBuilder = null; + if (parameterRangeTypeCase_ == 2) { + subBuilder = ((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_).toBuilder(); + } + parameterRangeType_ = + input.readMessage(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_); + parameterRangeType_ = subBuilder.buildPartial(); + } + parameterRangeTypeCase_ = 2; + break; + } + case 26: { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder subBuilder = null; + if (parameterRangeTypeCase_ == 3) { + subBuilder = ((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_).toBuilder(); + } + parameterRangeType_ = + input.readMessage(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_); + parameterRangeType_ = subBuilder.buildPartial(); + } + parameterRangeTypeCase_ = 3; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.Builder.class); + } + + private int parameterRangeTypeCase_ = 0; + private java.lang.Object parameterRangeType_; + public enum ParameterRangeTypeCase + implements com.google.protobuf.Internal.EnumLite { + CONTINUOUS_PARAMETER_RANGE(1), + INTEGER_PARAMETER_RANGE(2), + CATEGORICAL_PARAMETER_RANGE(3), + PARAMETERRANGETYPE_NOT_SET(0); + private final int value; + private ParameterRangeTypeCase(int value) { + this.value = value; + } + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static ParameterRangeTypeCase valueOf(int value) { + return forNumber(value); + } + + public static ParameterRangeTypeCase forNumber(int value) { + switch (value) { + case 1: return CONTINUOUS_PARAMETER_RANGE; + case 2: return INTEGER_PARAMETER_RANGE; + case 3: return CATEGORICAL_PARAMETER_RANGE; + case 0: return PARAMETERRANGETYPE_NOT_SET; + default: return null; + } + } + public int getNumber() { + return this.value; + } + }; + + public ParameterRangeTypeCase + getParameterRangeTypeCase() { + return ParameterRangeTypeCase.forNumber( + parameterRangeTypeCase_); + } + + public static final int CONTINUOUS_PARAMETER_RANGE_FIELD_NUMBER = 1; + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public boolean hasContinuousParameterRange() { + return parameterRangeTypeCase_ == 1; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getContinuousParameterRange() { + if (parameterRangeTypeCase_ == 1) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder getContinuousParameterRangeOrBuilder() { + if (parameterRangeTypeCase_ == 1) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + + public static final int INTEGER_PARAMETER_RANGE_FIELD_NUMBER = 2; + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public boolean hasIntegerParameterRange() { + return parameterRangeTypeCase_ == 2; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getIntegerParameterRange() { + if (parameterRangeTypeCase_ == 2) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder getIntegerParameterRangeOrBuilder() { + if (parameterRangeTypeCase_ == 2) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + + public static final int CATEGORICAL_PARAMETER_RANGE_FIELD_NUMBER = 3; + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public boolean hasCategoricalParameterRange() { + return parameterRangeTypeCase_ == 3; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getCategoricalParameterRange() { + if (parameterRangeTypeCase_ == 3) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder getCategoricalParameterRangeOrBuilder() { + if (parameterRangeTypeCase_ == 3) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (parameterRangeTypeCase_ == 1) { + output.writeMessage(1, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_); + } + if (parameterRangeTypeCase_ == 2) { + output.writeMessage(2, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_); + } + if (parameterRangeTypeCase_ == 3) { + output.writeMessage(3, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (parameterRangeTypeCase_ == 1) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(1, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_); + } + if (parameterRangeTypeCase_ == 2) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(2, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_); + } + if (parameterRangeTypeCase_ == 3) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(3, (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf) obj; + + if (!getParameterRangeTypeCase().equals(other.getParameterRangeTypeCase())) return false; + switch (parameterRangeTypeCase_) { + case 1: + if (!getContinuousParameterRange() + .equals(other.getContinuousParameterRange())) return false; + break; + case 2: + if (!getIntegerParameterRange() + .equals(other.getIntegerParameterRange())) return false; + break; + case 3: + if (!getCategoricalParameterRange() + .equals(other.getCategoricalParameterRange())) return false; + break; + case 0: + default: + } + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + switch (parameterRangeTypeCase_) { + case 1: + hash = (37 * hash) + CONTINUOUS_PARAMETER_RANGE_FIELD_NUMBER; + hash = (53 * hash) + getContinuousParameterRange().hashCode(); + break; + case 2: + hash = (37 * hash) + INTEGER_PARAMETER_RANGE_FIELD_NUMBER; + hash = (53 * hash) + getIntegerParameterRange().hashCode(); + break; + case 3: + hash = (37 * hash) + CATEGORICAL_PARAMETER_RANGE_FIELD_NUMBER; + hash = (53 * hash) + getCategoricalParameterRange().hashCode(); + break; + case 0: + default: + } + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * ParameterRangeOneOf describes a single ParameterRange, which is a one-of structure that can be one of
+     * the three possible types: ContinuousParameterRange, IntegerParameterRange, and CategoricalParameterRange.
+     * This one-of structure in Flyte enables specifying a Parameter in a type-safe manner
+     * See: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ParameterRangeOneOf} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOfOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf(this); + if (parameterRangeTypeCase_ == 1) { + if (continuousParameterRangeBuilder_ == null) { + result.parameterRangeType_ = parameterRangeType_; + } else { + result.parameterRangeType_ = continuousParameterRangeBuilder_.build(); + } + } + if (parameterRangeTypeCase_ == 2) { + if (integerParameterRangeBuilder_ == null) { + result.parameterRangeType_ = parameterRangeType_; + } else { + result.parameterRangeType_ = integerParameterRangeBuilder_.build(); + } + } + if (parameterRangeTypeCase_ == 3) { + if (categoricalParameterRangeBuilder_ == null) { + result.parameterRangeType_ = parameterRangeType_; + } else { + result.parameterRangeType_ = categoricalParameterRangeBuilder_.build(); + } + } + result.parameterRangeTypeCase_ = parameterRangeTypeCase_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.getDefaultInstance()) return this; + switch (other.getParameterRangeTypeCase()) { + case CONTINUOUS_PARAMETER_RANGE: { + mergeContinuousParameterRange(other.getContinuousParameterRange()); + break; + } + case INTEGER_PARAMETER_RANGE: { + mergeIntegerParameterRange(other.getIntegerParameterRange()); + break; + } + case CATEGORICAL_PARAMETER_RANGE: { + mergeCategoricalParameterRange(other.getCategoricalParameterRange()); + break; + } + case PARAMETERRANGETYPE_NOT_SET: { + break; + } + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + private int parameterRangeTypeCase_ = 0; + private java.lang.Object parameterRangeType_; + public ParameterRangeTypeCase + getParameterRangeTypeCase() { + return ParameterRangeTypeCase.forNumber( + parameterRangeTypeCase_); + } + + public Builder clearParameterRangeType() { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + onChanged(); + return this; + } + + + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder> continuousParameterRangeBuilder_; + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public boolean hasContinuousParameterRange() { + return parameterRangeTypeCase_ == 1; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange getContinuousParameterRange() { + if (continuousParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 1) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } else { + if (parameterRangeTypeCase_ == 1) { + return continuousParameterRangeBuilder_.getMessage(); + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public Builder setContinuousParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange value) { + if (continuousParameterRangeBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + parameterRangeType_ = value; + onChanged(); + } else { + continuousParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 1; + return this; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public Builder setContinuousParameterRange( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder builderForValue) { + if (continuousParameterRangeBuilder_ == null) { + parameterRangeType_ = builderForValue.build(); + onChanged(); + } else { + continuousParameterRangeBuilder_.setMessage(builderForValue.build()); + } + parameterRangeTypeCase_ = 1; + return this; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public Builder mergeContinuousParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange value) { + if (continuousParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 1 && + parameterRangeType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance()) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.newBuilder((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_) + .mergeFrom(value).buildPartial(); + } else { + parameterRangeType_ = value; + } + onChanged(); + } else { + if (parameterRangeTypeCase_ == 1) { + continuousParameterRangeBuilder_.mergeFrom(value); + } + continuousParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 1; + return this; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public Builder clearContinuousParameterRange() { + if (continuousParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 1) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + onChanged(); + } + } else { + if (parameterRangeTypeCase_ == 1) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + } + continuousParameterRangeBuilder_.clear(); + } + return this; + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder getContinuousParameterRangeBuilder() { + return getContinuousParameterRangeFieldBuilder().getBuilder(); + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder getContinuousParameterRangeOrBuilder() { + if ((parameterRangeTypeCase_ == 1) && (continuousParameterRangeBuilder_ != null)) { + return continuousParameterRangeBuilder_.getMessageOrBuilder(); + } else { + if (parameterRangeTypeCase_ == 1) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.ContinuousParameterRange continuous_parameter_range = 1; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder> + getContinuousParameterRangeFieldBuilder() { + if (continuousParameterRangeBuilder_ == null) { + if (!(parameterRangeTypeCase_ == 1)) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.getDefaultInstance(); + } + continuousParameterRangeBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRangeOrBuilder>( + (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ContinuousParameterRange) parameterRangeType_, + getParentForChildren(), + isClean()); + parameterRangeType_ = null; + } + parameterRangeTypeCase_ = 1; + onChanged();; + return continuousParameterRangeBuilder_; + } + + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder> integerParameterRangeBuilder_; + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public boolean hasIntegerParameterRange() { + return parameterRangeTypeCase_ == 2; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange getIntegerParameterRange() { + if (integerParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 2) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } else { + if (parameterRangeTypeCase_ == 2) { + return integerParameterRangeBuilder_.getMessage(); + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public Builder setIntegerParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange value) { + if (integerParameterRangeBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + parameterRangeType_ = value; + onChanged(); + } else { + integerParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 2; + return this; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public Builder setIntegerParameterRange( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder builderForValue) { + if (integerParameterRangeBuilder_ == null) { + parameterRangeType_ = builderForValue.build(); + onChanged(); + } else { + integerParameterRangeBuilder_.setMessage(builderForValue.build()); + } + parameterRangeTypeCase_ = 2; + return this; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public Builder mergeIntegerParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange value) { + if (integerParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 2 && + parameterRangeType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance()) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.newBuilder((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_) + .mergeFrom(value).buildPartial(); + } else { + parameterRangeType_ = value; + } + onChanged(); + } else { + if (parameterRangeTypeCase_ == 2) { + integerParameterRangeBuilder_.mergeFrom(value); + } + integerParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 2; + return this; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public Builder clearIntegerParameterRange() { + if (integerParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 2) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + onChanged(); + } + } else { + if (parameterRangeTypeCase_ == 2) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + } + integerParameterRangeBuilder_.clear(); + } + return this; + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder getIntegerParameterRangeBuilder() { + return getIntegerParameterRangeFieldBuilder().getBuilder(); + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder getIntegerParameterRangeOrBuilder() { + if ((parameterRangeTypeCase_ == 2) && (integerParameterRangeBuilder_ != null)) { + return integerParameterRangeBuilder_.getMessageOrBuilder(); + } else { + if (parameterRangeTypeCase_ == 2) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.IntegerParameterRange integer_parameter_range = 2; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder> + getIntegerParameterRangeFieldBuilder() { + if (integerParameterRangeBuilder_ == null) { + if (!(parameterRangeTypeCase_ == 2)) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.getDefaultInstance(); + } + integerParameterRangeBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRangeOrBuilder>( + (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.IntegerParameterRange) parameterRangeType_, + getParentForChildren(), + isClean()); + parameterRangeType_ = null; + } + parameterRangeTypeCase_ = 2; + onChanged();; + return integerParameterRangeBuilder_; + } + + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder> categoricalParameterRangeBuilder_; + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public boolean hasCategoricalParameterRange() { + return parameterRangeTypeCase_ == 3; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange getCategoricalParameterRange() { + if (categoricalParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 3) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } else { + if (parameterRangeTypeCase_ == 3) { + return categoricalParameterRangeBuilder_.getMessage(); + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public Builder setCategoricalParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange value) { + if (categoricalParameterRangeBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + parameterRangeType_ = value; + onChanged(); + } else { + categoricalParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 3; + return this; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public Builder setCategoricalParameterRange( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder builderForValue) { + if (categoricalParameterRangeBuilder_ == null) { + parameterRangeType_ = builderForValue.build(); + onChanged(); + } else { + categoricalParameterRangeBuilder_.setMessage(builderForValue.build()); + } + parameterRangeTypeCase_ = 3; + return this; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public Builder mergeCategoricalParameterRange(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange value) { + if (categoricalParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 3 && + parameterRangeType_ != flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance()) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.newBuilder((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_) + .mergeFrom(value).buildPartial(); + } else { + parameterRangeType_ = value; + } + onChanged(); + } else { + if (parameterRangeTypeCase_ == 3) { + categoricalParameterRangeBuilder_.mergeFrom(value); + } + categoricalParameterRangeBuilder_.setMessage(value); + } + parameterRangeTypeCase_ = 3; + return this; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public Builder clearCategoricalParameterRange() { + if (categoricalParameterRangeBuilder_ == null) { + if (parameterRangeTypeCase_ == 3) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + onChanged(); + } + } else { + if (parameterRangeTypeCase_ == 3) { + parameterRangeTypeCase_ = 0; + parameterRangeType_ = null; + } + categoricalParameterRangeBuilder_.clear(); + } + return this; + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder getCategoricalParameterRangeBuilder() { + return getCategoricalParameterRangeFieldBuilder().getBuilder(); + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder getCategoricalParameterRangeOrBuilder() { + if ((parameterRangeTypeCase_ == 3) && (categoricalParameterRangeBuilder_ != null)) { + return categoricalParameterRangeBuilder_.getMessageOrBuilder(); + } else { + if (parameterRangeTypeCase_ == 3) { + return (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_; + } + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + } + /** + * .flyteidl.plugins.sagemaker.CategoricalParameterRange categorical_parameter_range = 3; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder> + getCategoricalParameterRangeFieldBuilder() { + if (categoricalParameterRangeBuilder_ == null) { + if (!(parameterRangeTypeCase_ == 3)) { + parameterRangeType_ = flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.getDefaultInstance(); + } + categoricalParameterRangeBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange.Builder, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRangeOrBuilder>( + (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.CategoricalParameterRange) parameterRangeType_, + getParentForChildren(), + isClean()); + parameterRangeType_ = null; + } + parameterRangeTypeCase_ = 3; + onChanged();; + return categoricalParameterRangeBuilder_; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public ParameterRangeOneOf parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new ParameterRangeOneOf(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface ParameterRangesOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.ParameterRanges) + com.google.protobuf.MessageOrBuilder { + + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + int getParameterRangeMapCount(); + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + boolean containsParameterRangeMap( + java.lang.String key); + /** + * Use {@link #getParameterRangeMapMap()} instead. + */ + @java.lang.Deprecated + java.util.Map + getParameterRangeMap(); + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + java.util.Map + getParameterRangeMapMap(); + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrDefault( + java.lang.String key, + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf defaultValue); + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrThrow( + java.lang.String key); + } + /** + *
+   * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ParameterRanges} + */ + public static final class ParameterRanges extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.ParameterRanges) + ParameterRangesOrBuilder { + private static final long serialVersionUID = 0L; + // Use ParameterRanges.newBuilder() to construct. + private ParameterRanges(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private ParameterRanges() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private ParameterRanges( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + if (!((mutable_bitField0_ & 0x00000001) != 0)) { + parameterRangeMap_ = com.google.protobuf.MapField.newMapField( + ParameterRangeMapDefaultEntryHolder.defaultEntry); + mutable_bitField0_ |= 0x00000001; + } + com.google.protobuf.MapEntry + parameterRangeMap__ = input.readMessage( + ParameterRangeMapDefaultEntryHolder.defaultEntry.getParserForType(), extensionRegistry); + parameterRangeMap_.getMutableMap().put( + parameterRangeMap__.getKey(), parameterRangeMap__.getValue()); + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor; + } + + @SuppressWarnings({"rawtypes"}) + @java.lang.Override + protected com.google.protobuf.MapField internalGetMapField( + int number) { + switch (number) { + case 1: + return internalGetParameterRangeMap(); + default: + throw new RuntimeException( + "Invalid map field number: " + number); + } + } + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder.class); + } + + public static final int PARAMETER_RANGE_MAP_FIELD_NUMBER = 1; + private static final class ParameterRangeMapDefaultEntryHolder { + static final com.google.protobuf.MapEntry< + java.lang.String, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf> defaultEntry = + com.google.protobuf.MapEntry + .newDefaultInstance( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_descriptor, + com.google.protobuf.WireFormat.FieldType.STRING, + "", + com.google.protobuf.WireFormat.FieldType.MESSAGE, + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf.getDefaultInstance()); + } + private com.google.protobuf.MapField< + java.lang.String, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf> parameterRangeMap_; + private com.google.protobuf.MapField + internalGetParameterRangeMap() { + if (parameterRangeMap_ == null) { + return com.google.protobuf.MapField.emptyMapField( + ParameterRangeMapDefaultEntryHolder.defaultEntry); + } + return parameterRangeMap_; + } + + public int getParameterRangeMapCount() { + return internalGetParameterRangeMap().getMap().size(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public boolean containsParameterRangeMap( + java.lang.String key) { + if (key == null) { throw new java.lang.NullPointerException(); } + return internalGetParameterRangeMap().getMap().containsKey(key); + } + /** + * Use {@link #getParameterRangeMapMap()} instead. + */ + @java.lang.Deprecated + public java.util.Map getParameterRangeMap() { + return getParameterRangeMapMap(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public java.util.Map getParameterRangeMapMap() { + return internalGetParameterRangeMap().getMap(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrDefault( + java.lang.String key, + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf defaultValue) { + if (key == null) { throw new java.lang.NullPointerException(); } + java.util.Map map = + internalGetParameterRangeMap().getMap(); + return map.containsKey(key) ? map.get(key) : defaultValue; + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrThrow( + java.lang.String key) { + if (key == null) { throw new java.lang.NullPointerException(); } + java.util.Map map = + internalGetParameterRangeMap().getMap(); + if (!map.containsKey(key)) { + throw new java.lang.IllegalArgumentException(); + } + return map.get(key); + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + com.google.protobuf.GeneratedMessageV3 + .serializeStringMapTo( + output, + internalGetParameterRangeMap(), + ParameterRangeMapDefaultEntryHolder.defaultEntry, + 1); + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + for (java.util.Map.Entry entry + : internalGetParameterRangeMap().getMap().entrySet()) { + com.google.protobuf.MapEntry + parameterRangeMap__ = ParameterRangeMapDefaultEntryHolder.defaultEntry.newBuilderForType() + .setKey(entry.getKey()) + .setValue(entry.getValue()) + .build(); + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(1, parameterRangeMap__); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges other = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges) obj; + + if (!internalGetParameterRangeMap().equals( + other.internalGetParameterRangeMap())) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + if (!internalGetParameterRangeMap().getMap().isEmpty()) { + hash = (37 * hash) + PARAMETER_RANGE_MAP_FIELD_NUMBER; + hash = (53 * hash) + internalGetParameterRangeMap().hashCode(); + } + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.ParameterRanges} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.ParameterRanges) + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangesOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor; + } + + @SuppressWarnings({"rawtypes"}) + protected com.google.protobuf.MapField internalGetMapField( + int number) { + switch (number) { + case 1: + return internalGetParameterRangeMap(); + default: + throw new RuntimeException( + "Invalid map field number: " + number); + } + } + @SuppressWarnings({"rawtypes"}) + protected com.google.protobuf.MapField internalGetMutableMapField( + int number) { + switch (number) { + case 1: + return internalGetMutableParameterRangeMap(); + default: + throw new RuntimeException( + "Invalid map field number: " + number); + } + } + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.class, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + internalGetMutableParameterRangeMap().clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges build() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges buildPartial() { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges result = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges(this); + int from_bitField0_ = bitField0_; + result.parameterRangeMap_ = internalGetParameterRangeMap(); + result.parameterRangeMap_.makeImmutable(); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges) { + return mergeFrom((flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges other) { + if (other == flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges.getDefaultInstance()) return this; + internalGetMutableParameterRangeMap().mergeFrom( + other.internalGetParameterRangeMap()); + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + private int bitField0_; + + private com.google.protobuf.MapField< + java.lang.String, flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf> parameterRangeMap_; + private com.google.protobuf.MapField + internalGetParameterRangeMap() { + if (parameterRangeMap_ == null) { + return com.google.protobuf.MapField.emptyMapField( + ParameterRangeMapDefaultEntryHolder.defaultEntry); + } + return parameterRangeMap_; + } + private com.google.protobuf.MapField + internalGetMutableParameterRangeMap() { + onChanged();; + if (parameterRangeMap_ == null) { + parameterRangeMap_ = com.google.protobuf.MapField.newMapField( + ParameterRangeMapDefaultEntryHolder.defaultEntry); + } + if (!parameterRangeMap_.isMutable()) { + parameterRangeMap_ = parameterRangeMap_.copy(); + } + return parameterRangeMap_; + } + + public int getParameterRangeMapCount() { + return internalGetParameterRangeMap().getMap().size(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public boolean containsParameterRangeMap( + java.lang.String key) { + if (key == null) { throw new java.lang.NullPointerException(); } + return internalGetParameterRangeMap().getMap().containsKey(key); + } + /** + * Use {@link #getParameterRangeMapMap()} instead. + */ + @java.lang.Deprecated + public java.util.Map getParameterRangeMap() { + return getParameterRangeMapMap(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public java.util.Map getParameterRangeMapMap() { + return internalGetParameterRangeMap().getMap(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrDefault( + java.lang.String key, + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf defaultValue) { + if (key == null) { throw new java.lang.NullPointerException(); } + java.util.Map map = + internalGetParameterRangeMap().getMap(); + return map.containsKey(key) ? map.get(key) : defaultValue; + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf getParameterRangeMapOrThrow( + java.lang.String key) { + if (key == null) { throw new java.lang.NullPointerException(); } + java.util.Map map = + internalGetParameterRangeMap().getMap(); + if (!map.containsKey(key)) { + throw new java.lang.IllegalArgumentException(); + } + return map.get(key); + } + + public Builder clearParameterRangeMap() { + internalGetMutableParameterRangeMap().getMutableMap() + .clear(); + return this; + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public Builder removeParameterRangeMap( + java.lang.String key) { + if (key == null) { throw new java.lang.NullPointerException(); } + internalGetMutableParameterRangeMap().getMutableMap() + .remove(key); + return this; + } + /** + * Use alternate mutation accessors instead. + */ + @java.lang.Deprecated + public java.util.Map + getMutableParameterRangeMap() { + return internalGetMutableParameterRangeMap().getMutableMap(); + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + public Builder putParameterRangeMap( + java.lang.String key, + flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRangeOneOf value) { + if (key == null) { throw new java.lang.NullPointerException(); } + if (value == null) { throw new java.lang.NullPointerException(); } + internalGetMutableParameterRangeMap().getMutableMap() + .put(key, value); + return this; + } + /** + * map<string, .flyteidl.plugins.sagemaker.ParameterRangeOneOf> parameter_range_map = 1; + */ + + public Builder putAllParameterRangeMap( + java.util.Map values) { + internalGetMutableParameterRangeMap().getMutableMap() + .putAll(values); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.ParameterRanges) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRanges) + private static final flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges(); + } + + public static flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public ParameterRanges parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new ParameterRanges(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.ParameterRangesOuterClass.ParameterRanges getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_fieldAccessorTable; + + public static com.google.protobuf.Descriptors.FileDescriptor + getDescriptor() { + return descriptor; + } + private static com.google.protobuf.Descriptors.FileDescriptor + descriptor; + static { + java.lang.String[] descriptorData = { + "\n1flyteidl/plugins/sagemaker/parameter_r" + + "anges.proto\022\032flyteidl.plugins.sagemaker\"" + + "c\n\031HyperparameterScalingType\"F\n\005Value\022\010\n" + + "\004AUTO\020\000\022\n\n\006LINEAR\020\001\022\017\n\013LOGARITHMIC\020\002\022\026\n\022" + + "REVERSELOGARITHMIC\020\003\"\223\001\n\030ContinuousParam" + + "eterRange\022\021\n\tmax_value\030\001 \001(\001\022\021\n\tmin_valu" + + "e\030\002 \001(\001\022Q\n\014scaling_type\030\003 \001(\0162;.flyteidl" + + ".plugins.sagemaker.HyperparameterScaling" + + "Type.Value\"\220\001\n\025IntegerParameterRange\022\021\n\t" + + "max_value\030\001 \001(\003\022\021\n\tmin_value\030\002 \001(\003\022Q\n\014sc" + + "aling_type\030\003 \001(\0162;.flyteidl.plugins.sage" + + "maker.HyperparameterScalingType.Value\"+\n" + + "\031CategoricalParameterRange\022\016\n\006values\030\001 \003" + + "(\t\"\275\002\n\023ParameterRangeOneOf\022Z\n\032continuous" + + "_parameter_range\030\001 \001(\01324.flyteidl.plugin" + + "s.sagemaker.ContinuousParameterRangeH\000\022T" + + "\n\027integer_parameter_range\030\002 \001(\01321.flytei" + + "dl.plugins.sagemaker.IntegerParameterRan" + + "geH\000\022\\\n\033categorical_parameter_range\030\003 \001(" + + "\01325.flyteidl.plugins.sagemaker.Categoric" + + "alParameterRangeH\000B\026\n\024parameter_range_ty" + + "pe\"\335\001\n\017ParameterRanges\022_\n\023parameter_rang" + + "e_map\030\001 \003(\0132B.flyteidl.plugins.sagemaker" + + ".ParameterRanges.ParameterRangeMapEntry\032" + + "i\n\026ParameterRangeMapEntry\022\013\n\003key\030\001 \001(\t\022>" + + "\n\005value\030\002 \001(\0132/.flyteidl.plugins.sagemak" + + "er.ParameterRangeOneOf:\0028\001B5Z3github.com" + + "/lyft/flyteidl/gen/pb-go/flyteidl/plugin" + + "sb\006proto3" + }; + com.google.protobuf.Descriptors.FileDescriptor.InternalDescriptorAssigner assigner = + new com.google.protobuf.Descriptors.FileDescriptor. InternalDescriptorAssigner() { + public com.google.protobuf.ExtensionRegistry assignDescriptors( + com.google.protobuf.Descriptors.FileDescriptor root) { + descriptor = root; + return null; + } + }; + com.google.protobuf.Descriptors.FileDescriptor + .internalBuildGeneratedFileFrom(descriptorData, + new com.google.protobuf.Descriptors.FileDescriptor[] { + }, assigner); + internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor = + getDescriptor().getMessageTypes().get(0); + internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_HyperparameterScalingType_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor = + getDescriptor().getMessageTypes().get(1); + internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_ContinuousParameterRange_descriptor, + new java.lang.String[] { "MaxValue", "MinValue", "ScalingType", }); + internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor = + getDescriptor().getMessageTypes().get(2); + internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_IntegerParameterRange_descriptor, + new java.lang.String[] { "MaxValue", "MinValue", "ScalingType", }); + internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor = + getDescriptor().getMessageTypes().get(3); + internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_CategoricalParameterRange_descriptor, + new java.lang.String[] { "Values", }); + internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor = + getDescriptor().getMessageTypes().get(4); + internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_ParameterRangeOneOf_descriptor, + new java.lang.String[] { "ContinuousParameterRange", "IntegerParameterRange", "CategoricalParameterRange", "ParameterRangeType", }); + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor = + getDescriptor().getMessageTypes().get(5); + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor, + new java.lang.String[] { "ParameterRangeMap", }); + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_descriptor = + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_descriptor.getNestedTypes().get(0); + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_ParameterRanges_ParameterRangeMapEntry_descriptor, + new java.lang.String[] { "Key", "Value", }); + } + + // @@protoc_insertion_point(outer_class_scope) +} diff --git a/gen/pb-java/flyteidl/plugins/sagemaker/TrainingJobOuterClass.java b/gen/pb-java/flyteidl/plugins/sagemaker/TrainingJobOuterClass.java new file mode 100644 index 000000000..145062e99 --- /dev/null +++ b/gen/pb-java/flyteidl/plugins/sagemaker/TrainingJobOuterClass.java @@ -0,0 +1,5615 @@ +// Generated by the protocol buffer compiler. DO NOT EDIT! +// source: flyteidl/plugins/sagemaker/training_job.proto + +package flyteidl.plugins.sagemaker; + +public final class TrainingJobOuterClass { + private TrainingJobOuterClass() {} + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistryLite registry) { + } + + public static void registerAllExtensions( + com.google.protobuf.ExtensionRegistry registry) { + registerAllExtensions( + (com.google.protobuf.ExtensionRegistryLite) registry); + } + public interface InputModeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.InputMode) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * The input mode that the algorithm supports. When using the File input mode, SageMaker downloads
+   * the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker
+   * volume for training container. When using Pipe input mode, Amazon SageMaker streams data directly
+   * from S3 to the container.
+   * See: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+   * For the input modes that different SageMaker algorithms support, see:
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.InputMode} + */ + public static final class InputMode extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.InputMode) + InputModeOrBuilder { + private static final long serialVersionUID = 0L; + // Use InputMode.newBuilder() to construct. + private InputMode(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private InputMode() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private InputMode( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputMode_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.InputMode.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * FILE = 0; + */ + FILE(0), + /** + * PIPE = 1; + */ + PIPE(1), + UNRECOGNIZED(-1), + ; + + /** + * FILE = 0; + */ + public static final int FILE_VALUE = 0; + /** + * PIPE = 1; + */ + public static final int PIPE_VALUE = 1; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return FILE; + case 1: return PIPE; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.InputMode.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * The input mode that the algorithm supports. When using the File input mode, SageMaker downloads
+     * the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker
+     * volume for training container. When using Pipe input mode, Amazon SageMaker streams data directly
+     * from S3 to the container.
+     * See: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * For the input modes that different SageMaker algorithms support, see:
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.InputMode} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.InputMode) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputModeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputMode_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.InputMode) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputMode) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public InputMode parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new InputMode(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface AlgorithmNameOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.AlgorithmName) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * The algorithm name is used for deciding which pre-built image to point to.
+   * This is only required for use cases where SageMaker's built-in algorithm mode is used.
+   * While we currently only support a subset of the algorithms, more will be added to the list.
+   * See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.AlgorithmName} + */ + public static final class AlgorithmName extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.AlgorithmName) + AlgorithmNameOrBuilder { + private static final long serialVersionUID = 0L; + // Use AlgorithmName.newBuilder() to construct. + private AlgorithmName(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private AlgorithmName() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private AlgorithmName( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmName_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.AlgorithmName.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * CUSTOM = 0; + */ + CUSTOM(0), + /** + * XGBOOST = 1; + */ + XGBOOST(1), + UNRECOGNIZED(-1), + ; + + /** + * CUSTOM = 0; + */ + public static final int CUSTOM_VALUE = 0; + /** + * XGBOOST = 1; + */ + public static final int XGBOOST_VALUE = 1; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return CUSTOM; + case 1: return XGBOOST; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.AlgorithmName.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * The algorithm name is used for deciding which pre-built image to point to.
+     * This is only required for use cases where SageMaker's built-in algorithm mode is used.
+     * While we currently only support a subset of the algorithms, more will be added to the list.
+     * See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.AlgorithmName} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.AlgorithmName) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmNameOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmName_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.AlgorithmName) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmName) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public AlgorithmName parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new AlgorithmName(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface InputContentTypeOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.InputContentType) + com.google.protobuf.MessageOrBuilder { + } + /** + *
+   * Specifies the type of file for input data. Different SageMaker built-in algorithms require different file types of input data
+   * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+   * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.InputContentType} + */ + public static final class InputContentType extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.InputContentType) + InputContentTypeOrBuilder { + private static final long serialVersionUID = 0L; + // Use InputContentType.newBuilder() to construct. + private InputContentType(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private InputContentType() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private InputContentType( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputContentType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Builder.class); + } + + /** + * Protobuf enum {@code flyteidl.plugins.sagemaker.InputContentType.Value} + */ + public enum Value + implements com.google.protobuf.ProtocolMessageEnum { + /** + * TEXT_CSV = 0; + */ + TEXT_CSV(0), + UNRECOGNIZED(-1), + ; + + /** + * TEXT_CSV = 0; + */ + public static final int TEXT_CSV_VALUE = 0; + + + public final int getNumber() { + if (this == UNRECOGNIZED) { + throw new java.lang.IllegalArgumentException( + "Can't get the number of an unknown enum value."); + } + return value; + } + + /** + * @deprecated Use {@link #forNumber(int)} instead. + */ + @java.lang.Deprecated + public static Value valueOf(int value) { + return forNumber(value); + } + + public static Value forNumber(int value) { + switch (value) { + case 0: return TEXT_CSV; + default: return null; + } + } + + public static com.google.protobuf.Internal.EnumLiteMap + internalGetValueMap() { + return internalValueMap; + } + private static final com.google.protobuf.Internal.EnumLiteMap< + Value> internalValueMap = + new com.google.protobuf.Internal.EnumLiteMap() { + public Value findValueByNumber(int number) { + return Value.forNumber(number); + } + }; + + public final com.google.protobuf.Descriptors.EnumValueDescriptor + getValueDescriptor() { + return getDescriptor().getValues().get(ordinal()); + } + public final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptorForType() { + return getDescriptor(); + } + public static final com.google.protobuf.Descriptors.EnumDescriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.getDescriptor().getEnumTypes().get(0); + } + + private static final Value[] VALUES = values(); + + public static Value valueOf( + com.google.protobuf.Descriptors.EnumValueDescriptor desc) { + if (desc.getType() != getDescriptor()) { + throw new java.lang.IllegalArgumentException( + "EnumValueDescriptor is not for this type."); + } + if (desc.getIndex() == -1) { + return UNRECOGNIZED; + } + return VALUES[desc.getIndex()]; + } + + private final int value; + + private Value(int value) { + this.value = value; + } + + // @@protoc_insertion_point(enum_scope:flyteidl.plugins.sagemaker.InputContentType.Value) + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType) obj; + + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * Specifies the type of file for input data. Different SageMaker built-in algorithms require different file types of input data
+     * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.InputContentType} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.InputContentType) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentTypeOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputContentType_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType(this); + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.getDefaultInstance()) return this; + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.InputContentType) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputContentType) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public InputContentType parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new InputContentType(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface MetricDefinitionOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.MetricDefinition) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * User-defined name of the metric
+     * 
+ * + * string name = 1; + */ + java.lang.String getName(); + /** + *
+     * User-defined name of the metric
+     * 
+ * + * string name = 1; + */ + com.google.protobuf.ByteString + getNameBytes(); + + /** + *
+     * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+     * 
+ * + * string regex = 2; + */ + java.lang.String getRegex(); + /** + *
+     * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+     * 
+ * + * string regex = 2; + */ + com.google.protobuf.ByteString + getRegexBytes(); + } + /** + *
+   * Specifies a metric that the training algorithm writes to stderr or stdout.
+   * This object is a pass-through.
+   * See this for details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MetricDefinition.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.MetricDefinition} + */ + public static final class MetricDefinition extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.MetricDefinition) + MetricDefinitionOrBuilder { + private static final long serialVersionUID = 0L; + // Use MetricDefinition.newBuilder() to construct. + private MetricDefinition(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private MetricDefinition() { + name_ = ""; + regex_ = ""; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private MetricDefinition( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + java.lang.String s = input.readStringRequireUtf8(); + + name_ = s; + break; + } + case 18: { + java.lang.String s = input.readStringRequireUtf8(); + + regex_ = s; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_MetricDefinition_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder.class); + } + + public static final int NAME_FIELD_NUMBER = 1; + private volatile java.lang.Object name_; + /** + *
+     * User-defined name of the metric
+     * 
+ * + * string name = 1; + */ + public java.lang.String getName() { + java.lang.Object ref = name_; + if (ref instanceof java.lang.String) { + return (java.lang.String) ref; + } else { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + name_ = s; + return s; + } + } + /** + *
+     * User-defined name of the metric
+     * 
+ * + * string name = 1; + */ + public com.google.protobuf.ByteString + getNameBytes() { + java.lang.Object ref = name_; + if (ref instanceof java.lang.String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + name_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + + public static final int REGEX_FIELD_NUMBER = 2; + private volatile java.lang.Object regex_; + /** + *
+     * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+     * 
+ * + * string regex = 2; + */ + public java.lang.String getRegex() { + java.lang.Object ref = regex_; + if (ref instanceof java.lang.String) { + return (java.lang.String) ref; + } else { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + regex_ = s; + return s; + } + } + /** + *
+     * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+     * 
+ * + * string regex = 2; + */ + public com.google.protobuf.ByteString + getRegexBytes() { + java.lang.Object ref = regex_; + if (ref instanceof java.lang.String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + regex_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (!getNameBytes().isEmpty()) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 1, name_); + } + if (!getRegexBytes().isEmpty()) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 2, regex_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (!getNameBytes().isEmpty()) { + size += com.google.protobuf.GeneratedMessageV3.computeStringSize(1, name_); + } + if (!getRegexBytes().isEmpty()) { + size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, regex_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition) obj; + + if (!getName() + .equals(other.getName())) return false; + if (!getRegex() + .equals(other.getRegex())) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + NAME_FIELD_NUMBER; + hash = (53 * hash) + getName().hashCode(); + hash = (37 * hash) + REGEX_FIELD_NUMBER; + hash = (53 * hash) + getRegex().hashCode(); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * Specifies a metric that the training algorithm writes to stderr or stdout.
+     * This object is a pass-through.
+     * See this for details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MetricDefinition.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.MetricDefinition} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.MetricDefinition) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_MetricDefinition_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + name_ = ""; + + regex_ = ""; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition(this); + result.name_ = name_; + result.regex_ = regex_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.getDefaultInstance()) return this; + if (!other.getName().isEmpty()) { + name_ = other.name_; + onChanged(); + } + if (!other.getRegex().isEmpty()) { + regex_ = other.regex_; + onChanged(); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private java.lang.Object name_ = ""; + /** + *
+       * User-defined name of the metric
+       * 
+ * + * string name = 1; + */ + public java.lang.String getName() { + java.lang.Object ref = name_; + if (!(ref instanceof java.lang.String)) { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + name_ = s; + return s; + } else { + return (java.lang.String) ref; + } + } + /** + *
+       * User-defined name of the metric
+       * 
+ * + * string name = 1; + */ + public com.google.protobuf.ByteString + getNameBytes() { + java.lang.Object ref = name_; + if (ref instanceof String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + name_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + /** + *
+       * User-defined name of the metric
+       * 
+ * + * string name = 1; + */ + public Builder setName( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + + name_ = value; + onChanged(); + return this; + } + /** + *
+       * User-defined name of the metric
+       * 
+ * + * string name = 1; + */ + public Builder clearName() { + + name_ = getDefaultInstance().getName(); + onChanged(); + return this; + } + /** + *
+       * User-defined name of the metric
+       * 
+ * + * string name = 1; + */ + public Builder setNameBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + + name_ = value; + onChanged(); + return this; + } + + private java.lang.Object regex_ = ""; + /** + *
+       * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+       * 
+ * + * string regex = 2; + */ + public java.lang.String getRegex() { + java.lang.Object ref = regex_; + if (!(ref instanceof java.lang.String)) { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + regex_ = s; + return s; + } else { + return (java.lang.String) ref; + } + } + /** + *
+       * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+       * 
+ * + * string regex = 2; + */ + public com.google.protobuf.ByteString + getRegexBytes() { + java.lang.Object ref = regex_; + if (ref instanceof String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + regex_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + /** + *
+       * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+       * 
+ * + * string regex = 2; + */ + public Builder setRegex( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + + regex_ = value; + onChanged(); + return this; + } + /** + *
+       * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+       * 
+ * + * string regex = 2; + */ + public Builder clearRegex() { + + regex_ = getDefaultInstance().getRegex(); + onChanged(); + return this; + } + /** + *
+       * SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics
+       * 
+ * + * string regex = 2; + */ + public Builder setRegexBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + + regex_ = value; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.MetricDefinition) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.MetricDefinition) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public MetricDefinition parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new MetricDefinition(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface AlgorithmSpecificationOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.AlgorithmSpecification) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * The input mode can be either PIPE or FILE
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + int getInputModeValue(); + /** + *
+     * The input mode can be either PIPE or FILE
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value getInputMode(); + + /** + *
+     * The algorithm name is used for deciding which pre-built image to point to
+     * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + int getAlgorithmNameValue(); + /** + *
+     * The algorithm name is used for deciding which pre-built image to point to
+     * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value getAlgorithmName(); + + /** + *
+     * The algorithm version field is used for deciding which pre-built image to point to
+     * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+     * 
+ * + * string algorithm_version = 3; + */ + java.lang.String getAlgorithmVersion(); + /** + *
+     * The algorithm version field is used for deciding which pre-built image to point to
+     * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+     * 
+ * + * string algorithm_version = 3; + */ + com.google.protobuf.ByteString + getAlgorithmVersionBytes(); + + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + java.util.List + getMetricDefinitionsList(); + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getMetricDefinitions(int index); + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + int getMetricDefinitionsCount(); + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + java.util.List + getMetricDefinitionsOrBuilderList(); + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder getMetricDefinitionsOrBuilder( + int index); + + /** + *
+     * The content type of the input
+     * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + int getInputContentTypeValue(); + /** + *
+     * The content type of the input
+     * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value getInputContentType(); + } + /** + *
+   * Specifies the training algorithm to be used in the training job
+   * This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify
+   * TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm
+   * name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to
+   * CUSTOM. In this case, the value of the algorithm_version field has no effect
+   * For pass-through use cases: refer to this AWS official document for more details
+   * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.AlgorithmSpecification} + */ + public static final class AlgorithmSpecification extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.AlgorithmSpecification) + AlgorithmSpecificationOrBuilder { + private static final long serialVersionUID = 0L; + // Use AlgorithmSpecification.newBuilder() to construct. + private AlgorithmSpecification(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private AlgorithmSpecification() { + inputMode_ = 0; + algorithmName_ = 0; + algorithmVersion_ = ""; + metricDefinitions_ = java.util.Collections.emptyList(); + inputContentType_ = 0; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private AlgorithmSpecification( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 8: { + int rawValue = input.readEnum(); + + inputMode_ = rawValue; + break; + } + case 16: { + int rawValue = input.readEnum(); + + algorithmName_ = rawValue; + break; + } + case 26: { + java.lang.String s = input.readStringRequireUtf8(); + + algorithmVersion_ = s; + break; + } + case 34: { + if (!((mutable_bitField0_ & 0x00000008) != 0)) { + metricDefinitions_ = new java.util.ArrayList(); + mutable_bitField0_ |= 0x00000008; + } + metricDefinitions_.add( + input.readMessage(flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.parser(), extensionRegistry)); + break; + } + case 40: { + int rawValue = input.readEnum(); + + inputContentType_ = rawValue; + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + if (((mutable_bitField0_ & 0x00000008) != 0)) { + metricDefinitions_ = java.util.Collections.unmodifiableList(metricDefinitions_); + } + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder.class); + } + + private int bitField0_; + public static final int INPUT_MODE_FIELD_NUMBER = 1; + private int inputMode_; + /** + *
+     * The input mode can be either PIPE or FILE
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public int getInputModeValue() { + return inputMode_; + } + /** + *
+     * The input mode can be either PIPE or FILE
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value getInputMode() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.valueOf(inputMode_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.UNRECOGNIZED : result; + } + + public static final int ALGORITHM_NAME_FIELD_NUMBER = 2; + private int algorithmName_; + /** + *
+     * The algorithm name is used for deciding which pre-built image to point to
+     * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public int getAlgorithmNameValue() { + return algorithmName_; + } + /** + *
+     * The algorithm name is used for deciding which pre-built image to point to
+     * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value getAlgorithmName() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.valueOf(algorithmName_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.UNRECOGNIZED : result; + } + + public static final int ALGORITHM_VERSION_FIELD_NUMBER = 3; + private volatile java.lang.Object algorithmVersion_; + /** + *
+     * The algorithm version field is used for deciding which pre-built image to point to
+     * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+     * 
+ * + * string algorithm_version = 3; + */ + public java.lang.String getAlgorithmVersion() { + java.lang.Object ref = algorithmVersion_; + if (ref instanceof java.lang.String) { + return (java.lang.String) ref; + } else { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + algorithmVersion_ = s; + return s; + } + } + /** + *
+     * The algorithm version field is used for deciding which pre-built image to point to
+     * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+     * 
+ * + * string algorithm_version = 3; + */ + public com.google.protobuf.ByteString + getAlgorithmVersionBytes() { + java.lang.Object ref = algorithmVersion_; + if (ref instanceof java.lang.String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + algorithmVersion_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + + public static final int METRIC_DEFINITIONS_FIELD_NUMBER = 4; + private java.util.List metricDefinitions_; + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public java.util.List getMetricDefinitionsList() { + return metricDefinitions_; + } + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public java.util.List + getMetricDefinitionsOrBuilderList() { + return metricDefinitions_; + } + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public int getMetricDefinitionsCount() { + return metricDefinitions_.size(); + } + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getMetricDefinitions(int index) { + return metricDefinitions_.get(index); + } + /** + *
+     * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+     * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+     * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder getMetricDefinitionsOrBuilder( + int index) { + return metricDefinitions_.get(index); + } + + public static final int INPUT_CONTENT_TYPE_FIELD_NUMBER = 5; + private int inputContentType_; + /** + *
+     * The content type of the input
+     * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public int getInputContentTypeValue() { + return inputContentType_; + } + /** + *
+     * The content type of the input
+     * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+     * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+     * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value getInputContentType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.valueOf(inputContentType_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.UNRECOGNIZED : result; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (inputMode_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.FILE.getNumber()) { + output.writeEnum(1, inputMode_); + } + if (algorithmName_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.CUSTOM.getNumber()) { + output.writeEnum(2, algorithmName_); + } + if (!getAlgorithmVersionBytes().isEmpty()) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 3, algorithmVersion_); + } + for (int i = 0; i < metricDefinitions_.size(); i++) { + output.writeMessage(4, metricDefinitions_.get(i)); + } + if (inputContentType_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.TEXT_CSV.getNumber()) { + output.writeEnum(5, inputContentType_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (inputMode_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.FILE.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(1, inputMode_); + } + if (algorithmName_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.CUSTOM.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(2, algorithmName_); + } + if (!getAlgorithmVersionBytes().isEmpty()) { + size += com.google.protobuf.GeneratedMessageV3.computeStringSize(3, algorithmVersion_); + } + for (int i = 0; i < metricDefinitions_.size(); i++) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(4, metricDefinitions_.get(i)); + } + if (inputContentType_ != flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.TEXT_CSV.getNumber()) { + size += com.google.protobuf.CodedOutputStream + .computeEnumSize(5, inputContentType_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification) obj; + + if (inputMode_ != other.inputMode_) return false; + if (algorithmName_ != other.algorithmName_) return false; + if (!getAlgorithmVersion() + .equals(other.getAlgorithmVersion())) return false; + if (!getMetricDefinitionsList() + .equals(other.getMetricDefinitionsList())) return false; + if (inputContentType_ != other.inputContentType_) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + INPUT_MODE_FIELD_NUMBER; + hash = (53 * hash) + inputMode_; + hash = (37 * hash) + ALGORITHM_NAME_FIELD_NUMBER; + hash = (53 * hash) + algorithmName_; + hash = (37 * hash) + ALGORITHM_VERSION_FIELD_NUMBER; + hash = (53 * hash) + getAlgorithmVersion().hashCode(); + if (getMetricDefinitionsCount() > 0) { + hash = (37 * hash) + METRIC_DEFINITIONS_FIELD_NUMBER; + hash = (53 * hash) + getMetricDefinitionsList().hashCode(); + } + hash = (37 * hash) + INPUT_CONTENT_TYPE_FIELD_NUMBER; + hash = (53 * hash) + inputContentType_; + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * Specifies the training algorithm to be used in the training job
+     * This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify
+     * TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm
+     * name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to
+     * CUSTOM. In this case, the value of the algorithm_version field has no effect
+     * For pass-through use cases: refer to this AWS official document for more details
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.AlgorithmSpecification} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.AlgorithmSpecification) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + getMetricDefinitionsFieldBuilder(); + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + inputMode_ = 0; + + algorithmName_ = 0; + + algorithmVersion_ = ""; + + if (metricDefinitionsBuilder_ == null) { + metricDefinitions_ = java.util.Collections.emptyList(); + bitField0_ = (bitField0_ & ~0x00000008); + } else { + metricDefinitionsBuilder_.clear(); + } + inputContentType_ = 0; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification(this); + int from_bitField0_ = bitField0_; + int to_bitField0_ = 0; + result.inputMode_ = inputMode_; + result.algorithmName_ = algorithmName_; + result.algorithmVersion_ = algorithmVersion_; + if (metricDefinitionsBuilder_ == null) { + if (((bitField0_ & 0x00000008) != 0)) { + metricDefinitions_ = java.util.Collections.unmodifiableList(metricDefinitions_); + bitField0_ = (bitField0_ & ~0x00000008); + } + result.metricDefinitions_ = metricDefinitions_; + } else { + result.metricDefinitions_ = metricDefinitionsBuilder_.build(); + } + result.inputContentType_ = inputContentType_; + result.bitField0_ = to_bitField0_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.getDefaultInstance()) return this; + if (other.inputMode_ != 0) { + setInputModeValue(other.getInputModeValue()); + } + if (other.algorithmName_ != 0) { + setAlgorithmNameValue(other.getAlgorithmNameValue()); + } + if (!other.getAlgorithmVersion().isEmpty()) { + algorithmVersion_ = other.algorithmVersion_; + onChanged(); + } + if (metricDefinitionsBuilder_ == null) { + if (!other.metricDefinitions_.isEmpty()) { + if (metricDefinitions_.isEmpty()) { + metricDefinitions_ = other.metricDefinitions_; + bitField0_ = (bitField0_ & ~0x00000008); + } else { + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.addAll(other.metricDefinitions_); + } + onChanged(); + } + } else { + if (!other.metricDefinitions_.isEmpty()) { + if (metricDefinitionsBuilder_.isEmpty()) { + metricDefinitionsBuilder_.dispose(); + metricDefinitionsBuilder_ = null; + metricDefinitions_ = other.metricDefinitions_; + bitField0_ = (bitField0_ & ~0x00000008); + metricDefinitionsBuilder_ = + com.google.protobuf.GeneratedMessageV3.alwaysUseFieldBuilders ? + getMetricDefinitionsFieldBuilder() : null; + } else { + metricDefinitionsBuilder_.addAllMessages(other.metricDefinitions_); + } + } + } + if (other.inputContentType_ != 0) { + setInputContentTypeValue(other.getInputContentTypeValue()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + private int bitField0_; + + private int inputMode_ = 0; + /** + *
+       * The input mode can be either PIPE or FILE
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public int getInputModeValue() { + return inputMode_; + } + /** + *
+       * The input mode can be either PIPE or FILE
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public Builder setInputModeValue(int value) { + inputMode_ = value; + onChanged(); + return this; + } + /** + *
+       * The input mode can be either PIPE or FILE
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value getInputMode() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.valueOf(inputMode_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value.UNRECOGNIZED : result; + } + /** + *
+       * The input mode can be either PIPE or FILE
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public Builder setInputMode(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputMode.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + inputMode_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * The input mode can be either PIPE or FILE
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputMode.Value input_mode = 1; + */ + public Builder clearInputMode() { + + inputMode_ = 0; + onChanged(); + return this; + } + + private int algorithmName_ = 0; + /** + *
+       * The algorithm name is used for deciding which pre-built image to point to
+       * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public int getAlgorithmNameValue() { + return algorithmName_; + } + /** + *
+       * The algorithm name is used for deciding which pre-built image to point to
+       * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public Builder setAlgorithmNameValue(int value) { + algorithmName_ = value; + onChanged(); + return this; + } + /** + *
+       * The algorithm name is used for deciding which pre-built image to point to
+       * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value getAlgorithmName() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.valueOf(algorithmName_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value.UNRECOGNIZED : result; + } + /** + *
+       * The algorithm name is used for deciding which pre-built image to point to
+       * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public Builder setAlgorithmName(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmName.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + algorithmName_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * The algorithm name is used for deciding which pre-built image to point to
+       * 
+ * + * .flyteidl.plugins.sagemaker.AlgorithmName.Value algorithm_name = 2; + */ + public Builder clearAlgorithmName() { + + algorithmName_ = 0; + onChanged(); + return this; + } + + private java.lang.Object algorithmVersion_ = ""; + /** + *
+       * The algorithm version field is used for deciding which pre-built image to point to
+       * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+       * 
+ * + * string algorithm_version = 3; + */ + public java.lang.String getAlgorithmVersion() { + java.lang.Object ref = algorithmVersion_; + if (!(ref instanceof java.lang.String)) { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + algorithmVersion_ = s; + return s; + } else { + return (java.lang.String) ref; + } + } + /** + *
+       * The algorithm version field is used for deciding which pre-built image to point to
+       * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+       * 
+ * + * string algorithm_version = 3; + */ + public com.google.protobuf.ByteString + getAlgorithmVersionBytes() { + java.lang.Object ref = algorithmVersion_; + if (ref instanceof String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + algorithmVersion_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + /** + *
+       * The algorithm version field is used for deciding which pre-built image to point to
+       * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+       * 
+ * + * string algorithm_version = 3; + */ + public Builder setAlgorithmVersion( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + + algorithmVersion_ = value; + onChanged(); + return this; + } + /** + *
+       * The algorithm version field is used for deciding which pre-built image to point to
+       * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+       * 
+ * + * string algorithm_version = 3; + */ + public Builder clearAlgorithmVersion() { + + algorithmVersion_ = getDefaultInstance().getAlgorithmVersion(); + onChanged(); + return this; + } + /** + *
+       * The algorithm version field is used for deciding which pre-built image to point to
+       * This is only needed for use cases where SageMaker's built-in algorithm mode is chosen
+       * 
+ * + * string algorithm_version = 3; + */ + public Builder setAlgorithmVersionBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + + algorithmVersion_ = value; + onChanged(); + return this; + } + + private java.util.List metricDefinitions_ = + java.util.Collections.emptyList(); + private void ensureMetricDefinitionsIsMutable() { + if (!((bitField0_ & 0x00000008) != 0)) { + metricDefinitions_ = new java.util.ArrayList(metricDefinitions_); + bitField0_ |= 0x00000008; + } + } + + private com.google.protobuf.RepeatedFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder> metricDefinitionsBuilder_; + + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public java.util.List getMetricDefinitionsList() { + if (metricDefinitionsBuilder_ == null) { + return java.util.Collections.unmodifiableList(metricDefinitions_); + } else { + return metricDefinitionsBuilder_.getMessageList(); + } + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public int getMetricDefinitionsCount() { + if (metricDefinitionsBuilder_ == null) { + return metricDefinitions_.size(); + } else { + return metricDefinitionsBuilder_.getCount(); + } + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition getMetricDefinitions(int index) { + if (metricDefinitionsBuilder_ == null) { + return metricDefinitions_.get(index); + } else { + return metricDefinitionsBuilder_.getMessage(index); + } + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder setMetricDefinitions( + int index, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition value) { + if (metricDefinitionsBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.set(index, value); + onChanged(); + } else { + metricDefinitionsBuilder_.setMessage(index, value); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder setMetricDefinitions( + int index, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder builderForValue) { + if (metricDefinitionsBuilder_ == null) { + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.set(index, builderForValue.build()); + onChanged(); + } else { + metricDefinitionsBuilder_.setMessage(index, builderForValue.build()); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder addMetricDefinitions(flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition value) { + if (metricDefinitionsBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.add(value); + onChanged(); + } else { + metricDefinitionsBuilder_.addMessage(value); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder addMetricDefinitions( + int index, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition value) { + if (metricDefinitionsBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.add(index, value); + onChanged(); + } else { + metricDefinitionsBuilder_.addMessage(index, value); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder addMetricDefinitions( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder builderForValue) { + if (metricDefinitionsBuilder_ == null) { + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.add(builderForValue.build()); + onChanged(); + } else { + metricDefinitionsBuilder_.addMessage(builderForValue.build()); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder addMetricDefinitions( + int index, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder builderForValue) { + if (metricDefinitionsBuilder_ == null) { + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.add(index, builderForValue.build()); + onChanged(); + } else { + metricDefinitionsBuilder_.addMessage(index, builderForValue.build()); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder addAllMetricDefinitions( + java.lang.Iterable values) { + if (metricDefinitionsBuilder_ == null) { + ensureMetricDefinitionsIsMutable(); + com.google.protobuf.AbstractMessageLite.Builder.addAll( + values, metricDefinitions_); + onChanged(); + } else { + metricDefinitionsBuilder_.addAllMessages(values); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder clearMetricDefinitions() { + if (metricDefinitionsBuilder_ == null) { + metricDefinitions_ = java.util.Collections.emptyList(); + bitField0_ = (bitField0_ & ~0x00000008); + onChanged(); + } else { + metricDefinitionsBuilder_.clear(); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public Builder removeMetricDefinitions(int index) { + if (metricDefinitionsBuilder_ == null) { + ensureMetricDefinitionsIsMutable(); + metricDefinitions_.remove(index); + onChanged(); + } else { + metricDefinitionsBuilder_.remove(index); + } + return this; + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder getMetricDefinitionsBuilder( + int index) { + return getMetricDefinitionsFieldBuilder().getBuilder(index); + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder getMetricDefinitionsOrBuilder( + int index) { + if (metricDefinitionsBuilder_ == null) { + return metricDefinitions_.get(index); } else { + return metricDefinitionsBuilder_.getMessageOrBuilder(index); + } + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public java.util.List + getMetricDefinitionsOrBuilderList() { + if (metricDefinitionsBuilder_ != null) { + return metricDefinitionsBuilder_.getMessageOrBuilderList(); + } else { + return java.util.Collections.unmodifiableList(metricDefinitions_); + } + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder addMetricDefinitionsBuilder() { + return getMetricDefinitionsFieldBuilder().addBuilder( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.getDefaultInstance()); + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder addMetricDefinitionsBuilder( + int index) { + return getMetricDefinitionsFieldBuilder().addBuilder( + index, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.getDefaultInstance()); + } + /** + *
+       * A list of metric definitions for SageMaker to evaluate/track on the progress of the training job
+       * See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html
+       * and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html
+       * 
+ * + * repeated .flyteidl.plugins.sagemaker.MetricDefinition metric_definitions = 4; + */ + public java.util.List + getMetricDefinitionsBuilderList() { + return getMetricDefinitionsFieldBuilder().getBuilderList(); + } + private com.google.protobuf.RepeatedFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder> + getMetricDefinitionsFieldBuilder() { + if (metricDefinitionsBuilder_ == null) { + metricDefinitionsBuilder_ = new com.google.protobuf.RepeatedFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinition.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.MetricDefinitionOrBuilder>( + metricDefinitions_, + ((bitField0_ & 0x00000008) != 0), + getParentForChildren(), + isClean()); + metricDefinitions_ = null; + } + return metricDefinitionsBuilder_; + } + + private int inputContentType_ = 0; + /** + *
+       * The content type of the input
+       * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+       * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public int getInputContentTypeValue() { + return inputContentType_; + } + /** + *
+       * The content type of the input
+       * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+       * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public Builder setInputContentTypeValue(int value) { + inputContentType_ = value; + onChanged(); + return this; + } + /** + *
+       * The content type of the input
+       * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+       * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value getInputContentType() { + @SuppressWarnings("deprecation") + flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value result = flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.valueOf(inputContentType_); + return result == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value.UNRECOGNIZED : result; + } + /** + *
+       * The content type of the input
+       * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+       * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public Builder setInputContentType(flyteidl.plugins.sagemaker.TrainingJobOuterClass.InputContentType.Value value) { + if (value == null) { + throw new NullPointerException(); + } + + inputContentType_ = value.getNumber(); + onChanged(); + return this; + } + /** + *
+       * The content type of the input
+       * See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html
+       * https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html
+       * 
+ * + * .flyteidl.plugins.sagemaker.InputContentType.Value input_content_type = 5; + */ + public Builder clearInputContentType() { + + inputContentType_ = 0; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.AlgorithmSpecification) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmSpecification) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public AlgorithmSpecification parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new AlgorithmSpecification(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface TrainingJobResourceConfigOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + com.google.protobuf.MessageOrBuilder { + + /** + *
+     * The number of ML compute instances to use. For distributed training, provide a value greater than 1.
+     * 
+ * + * int64 instance_count = 1; + */ + long getInstanceCount(); + + /** + *
+     * The ML compute instance type
+     * 
+ * + * string instance_type = 2; + */ + java.lang.String getInstanceType(); + /** + *
+     * The ML compute instance type
+     * 
+ * + * string instance_type = 2; + */ + com.google.protobuf.ByteString + getInstanceTypeBytes(); + + /** + *
+     * The size of the ML storage volume that you want to provision.
+     * 
+ * + * int64 volume_size_in_gb = 3; + */ + long getVolumeSizeInGb(); + } + /** + *
+   * TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the
+   * number of instances to launch, and the size of the ML storage volume the user wants to provision
+   * Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJobResourceConfig} + */ + public static final class TrainingJobResourceConfig extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + TrainingJobResourceConfigOrBuilder { + private static final long serialVersionUID = 0L; + // Use TrainingJobResourceConfig.newBuilder() to construct. + private TrainingJobResourceConfig(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private TrainingJobResourceConfig() { + instanceType_ = ""; + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private TrainingJobResourceConfig( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 8: { + + instanceCount_ = input.readInt64(); + break; + } + case 18: { + java.lang.String s = input.readStringRequireUtf8(); + + instanceType_ = s; + break; + } + case 24: { + + volumeSizeInGb_ = input.readInt64(); + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder.class); + } + + public static final int INSTANCE_COUNT_FIELD_NUMBER = 1; + private long instanceCount_; + /** + *
+     * The number of ML compute instances to use. For distributed training, provide a value greater than 1.
+     * 
+ * + * int64 instance_count = 1; + */ + public long getInstanceCount() { + return instanceCount_; + } + + public static final int INSTANCE_TYPE_FIELD_NUMBER = 2; + private volatile java.lang.Object instanceType_; + /** + *
+     * The ML compute instance type
+     * 
+ * + * string instance_type = 2; + */ + public java.lang.String getInstanceType() { + java.lang.Object ref = instanceType_; + if (ref instanceof java.lang.String) { + return (java.lang.String) ref; + } else { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + instanceType_ = s; + return s; + } + } + /** + *
+     * The ML compute instance type
+     * 
+ * + * string instance_type = 2; + */ + public com.google.protobuf.ByteString + getInstanceTypeBytes() { + java.lang.Object ref = instanceType_; + if (ref instanceof java.lang.String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + instanceType_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + + public static final int VOLUME_SIZE_IN_GB_FIELD_NUMBER = 3; + private long volumeSizeInGb_; + /** + *
+     * The size of the ML storage volume that you want to provision.
+     * 
+ * + * int64 volume_size_in_gb = 3; + */ + public long getVolumeSizeInGb() { + return volumeSizeInGb_; + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (instanceCount_ != 0L) { + output.writeInt64(1, instanceCount_); + } + if (!getInstanceTypeBytes().isEmpty()) { + com.google.protobuf.GeneratedMessageV3.writeString(output, 2, instanceType_); + } + if (volumeSizeInGb_ != 0L) { + output.writeInt64(3, volumeSizeInGb_); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (instanceCount_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(1, instanceCount_); + } + if (!getInstanceTypeBytes().isEmpty()) { + size += com.google.protobuf.GeneratedMessageV3.computeStringSize(2, instanceType_); + } + if (volumeSizeInGb_ != 0L) { + size += com.google.protobuf.CodedOutputStream + .computeInt64Size(3, volumeSizeInGb_); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig) obj; + + if (getInstanceCount() + != other.getInstanceCount()) return false; + if (!getInstanceType() + .equals(other.getInstanceType())) return false; + if (getVolumeSizeInGb() + != other.getVolumeSizeInGb()) return false; + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + hash = (37 * hash) + INSTANCE_COUNT_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getInstanceCount()); + hash = (37 * hash) + INSTANCE_TYPE_FIELD_NUMBER; + hash = (53 * hash) + getInstanceType().hashCode(); + hash = (37 * hash) + VOLUME_SIZE_IN_GB_FIELD_NUMBER; + hash = (53 * hash) + com.google.protobuf.Internal.hashLong( + getVolumeSizeInGb()); + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the
+     * number of instances to launch, and the size of the ML storage volume the user wants to provision
+     * Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJobResourceConfig} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + instanceCount_ = 0L; + + instanceType_ = ""; + + volumeSizeInGb_ = 0L; + + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig(this); + result.instanceCount_ = instanceCount_; + result.instanceType_ = instanceType_; + result.volumeSizeInGb_ = volumeSizeInGb_; + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.getDefaultInstance()) return this; + if (other.getInstanceCount() != 0L) { + setInstanceCount(other.getInstanceCount()); + } + if (!other.getInstanceType().isEmpty()) { + instanceType_ = other.instanceType_; + onChanged(); + } + if (other.getVolumeSizeInGb() != 0L) { + setVolumeSizeInGb(other.getVolumeSizeInGb()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private long instanceCount_ ; + /** + *
+       * The number of ML compute instances to use. For distributed training, provide a value greater than 1.
+       * 
+ * + * int64 instance_count = 1; + */ + public long getInstanceCount() { + return instanceCount_; + } + /** + *
+       * The number of ML compute instances to use. For distributed training, provide a value greater than 1.
+       * 
+ * + * int64 instance_count = 1; + */ + public Builder setInstanceCount(long value) { + + instanceCount_ = value; + onChanged(); + return this; + } + /** + *
+       * The number of ML compute instances to use. For distributed training, provide a value greater than 1.
+       * 
+ * + * int64 instance_count = 1; + */ + public Builder clearInstanceCount() { + + instanceCount_ = 0L; + onChanged(); + return this; + } + + private java.lang.Object instanceType_ = ""; + /** + *
+       * The ML compute instance type
+       * 
+ * + * string instance_type = 2; + */ + public java.lang.String getInstanceType() { + java.lang.Object ref = instanceType_; + if (!(ref instanceof java.lang.String)) { + com.google.protobuf.ByteString bs = + (com.google.protobuf.ByteString) ref; + java.lang.String s = bs.toStringUtf8(); + instanceType_ = s; + return s; + } else { + return (java.lang.String) ref; + } + } + /** + *
+       * The ML compute instance type
+       * 
+ * + * string instance_type = 2; + */ + public com.google.protobuf.ByteString + getInstanceTypeBytes() { + java.lang.Object ref = instanceType_; + if (ref instanceof String) { + com.google.protobuf.ByteString b = + com.google.protobuf.ByteString.copyFromUtf8( + (java.lang.String) ref); + instanceType_ = b; + return b; + } else { + return (com.google.protobuf.ByteString) ref; + } + } + /** + *
+       * The ML compute instance type
+       * 
+ * + * string instance_type = 2; + */ + public Builder setInstanceType( + java.lang.String value) { + if (value == null) { + throw new NullPointerException(); + } + + instanceType_ = value; + onChanged(); + return this; + } + /** + *
+       * The ML compute instance type
+       * 
+ * + * string instance_type = 2; + */ + public Builder clearInstanceType() { + + instanceType_ = getDefaultInstance().getInstanceType(); + onChanged(); + return this; + } + /** + *
+       * The ML compute instance type
+       * 
+ * + * string instance_type = 2; + */ + public Builder setInstanceTypeBytes( + com.google.protobuf.ByteString value) { + if (value == null) { + throw new NullPointerException(); + } + checkByteStringIsUtf8(value); + + instanceType_ = value; + onChanged(); + return this; + } + + private long volumeSizeInGb_ ; + /** + *
+       * The size of the ML storage volume that you want to provision.
+       * 
+ * + * int64 volume_size_in_gb = 3; + */ + public long getVolumeSizeInGb() { + return volumeSizeInGb_; + } + /** + *
+       * The size of the ML storage volume that you want to provision.
+       * 
+ * + * int64 volume_size_in_gb = 3; + */ + public Builder setVolumeSizeInGb(long value) { + + volumeSizeInGb_ = value; + onChanged(); + return this; + } + /** + *
+       * The size of the ML storage volume that you want to provision.
+       * 
+ * + * int64 volume_size_in_gb = 3; + */ + public Builder clearVolumeSizeInGb() { + + volumeSizeInGb_ = 0L; + onChanged(); + return this; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public TrainingJobResourceConfig parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new TrainingJobResourceConfig(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + public interface TrainingJobOrBuilder extends + // @@protoc_insertion_point(interface_extends:flyteidl.plugins.sagemaker.TrainingJob) + com.google.protobuf.MessageOrBuilder { + + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + boolean hasAlgorithmSpecification(); + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getAlgorithmSpecification(); + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder getAlgorithmSpecificationOrBuilder(); + + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + boolean hasTrainingJobResourceConfig(); + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getTrainingJobResourceConfig(); + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder getTrainingJobResourceConfigOrBuilder(); + } + /** + *
+   * The spec of a training job. This is mostly a pass-through object
+   * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html
+   * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJob} + */ + public static final class TrainingJob extends + com.google.protobuf.GeneratedMessageV3 implements + // @@protoc_insertion_point(message_implements:flyteidl.plugins.sagemaker.TrainingJob) + TrainingJobOrBuilder { + private static final long serialVersionUID = 0L; + // Use TrainingJob.newBuilder() to construct. + private TrainingJob(com.google.protobuf.GeneratedMessageV3.Builder builder) { + super(builder); + } + private TrainingJob() { + } + + @java.lang.Override + public final com.google.protobuf.UnknownFieldSet + getUnknownFields() { + return this.unknownFields; + } + private TrainingJob( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + this(); + if (extensionRegistry == null) { + throw new java.lang.NullPointerException(); + } + int mutable_bitField0_ = 0; + com.google.protobuf.UnknownFieldSet.Builder unknownFields = + com.google.protobuf.UnknownFieldSet.newBuilder(); + try { + boolean done = false; + while (!done) { + int tag = input.readTag(); + switch (tag) { + case 0: + done = true; + break; + case 10: { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder subBuilder = null; + if (algorithmSpecification_ != null) { + subBuilder = algorithmSpecification_.toBuilder(); + } + algorithmSpecification_ = input.readMessage(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom(algorithmSpecification_); + algorithmSpecification_ = subBuilder.buildPartial(); + } + + break; + } + case 18: { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder subBuilder = null; + if (trainingJobResourceConfig_ != null) { + subBuilder = trainingJobResourceConfig_.toBuilder(); + } + trainingJobResourceConfig_ = input.readMessage(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.parser(), extensionRegistry); + if (subBuilder != null) { + subBuilder.mergeFrom(trainingJobResourceConfig_); + trainingJobResourceConfig_ = subBuilder.buildPartial(); + } + + break; + } + default: { + if (!parseUnknownField( + input, unknownFields, extensionRegistry, tag)) { + done = true; + } + break; + } + } + } + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + throw e.setUnfinishedMessage(this); + } catch (java.io.IOException e) { + throw new com.google.protobuf.InvalidProtocolBufferException( + e).setUnfinishedMessage(this); + } finally { + this.unknownFields = unknownFields.build(); + makeExtensionsImmutable(); + } + } + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJob_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder.class); + } + + public static final int ALGORITHM_SPECIFICATION_FIELD_NUMBER = 1; + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification algorithmSpecification_; + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public boolean hasAlgorithmSpecification() { + return algorithmSpecification_ != null; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getAlgorithmSpecification() { + return algorithmSpecification_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.getDefaultInstance() : algorithmSpecification_; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder getAlgorithmSpecificationOrBuilder() { + return getAlgorithmSpecification(); + } + + public static final int TRAINING_JOB_RESOURCE_CONFIG_FIELD_NUMBER = 2; + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig trainingJobResourceConfig_; + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public boolean hasTrainingJobResourceConfig() { + return trainingJobResourceConfig_ != null; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getTrainingJobResourceConfig() { + return trainingJobResourceConfig_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.getDefaultInstance() : trainingJobResourceConfig_; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder getTrainingJobResourceConfigOrBuilder() { + return getTrainingJobResourceConfig(); + } + + private byte memoizedIsInitialized = -1; + @java.lang.Override + public final boolean isInitialized() { + byte isInitialized = memoizedIsInitialized; + if (isInitialized == 1) return true; + if (isInitialized == 0) return false; + + memoizedIsInitialized = 1; + return true; + } + + @java.lang.Override + public void writeTo(com.google.protobuf.CodedOutputStream output) + throws java.io.IOException { + if (algorithmSpecification_ != null) { + output.writeMessage(1, getAlgorithmSpecification()); + } + if (trainingJobResourceConfig_ != null) { + output.writeMessage(2, getTrainingJobResourceConfig()); + } + unknownFields.writeTo(output); + } + + @java.lang.Override + public int getSerializedSize() { + int size = memoizedSize; + if (size != -1) return size; + + size = 0; + if (algorithmSpecification_ != null) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(1, getAlgorithmSpecification()); + } + if (trainingJobResourceConfig_ != null) { + size += com.google.protobuf.CodedOutputStream + .computeMessageSize(2, getTrainingJobResourceConfig()); + } + size += unknownFields.getSerializedSize(); + memoizedSize = size; + return size; + } + + @java.lang.Override + public boolean equals(final java.lang.Object obj) { + if (obj == this) { + return true; + } + if (!(obj instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob)) { + return super.equals(obj); + } + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob other = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob) obj; + + if (hasAlgorithmSpecification() != other.hasAlgorithmSpecification()) return false; + if (hasAlgorithmSpecification()) { + if (!getAlgorithmSpecification() + .equals(other.getAlgorithmSpecification())) return false; + } + if (hasTrainingJobResourceConfig() != other.hasTrainingJobResourceConfig()) return false; + if (hasTrainingJobResourceConfig()) { + if (!getTrainingJobResourceConfig() + .equals(other.getTrainingJobResourceConfig())) return false; + } + if (!unknownFields.equals(other.unknownFields)) return false; + return true; + } + + @java.lang.Override + public int hashCode() { + if (memoizedHashCode != 0) { + return memoizedHashCode; + } + int hash = 41; + hash = (19 * hash) + getDescriptor().hashCode(); + if (hasAlgorithmSpecification()) { + hash = (37 * hash) + ALGORITHM_SPECIFICATION_FIELD_NUMBER; + hash = (53 * hash) + getAlgorithmSpecification().hashCode(); + } + if (hasTrainingJobResourceConfig()) { + hash = (37 * hash) + TRAINING_JOB_RESOURCE_CONFIG_FIELD_NUMBER; + hash = (53 * hash) + getTrainingJobResourceConfig().hashCode(); + } + hash = (29 * hash) + unknownFields.hashCode(); + memoizedHashCode = hash; + return hash; + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + java.nio.ByteBuffer data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + java.nio.ByteBuffer data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + com.google.protobuf.ByteString data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + com.google.protobuf.ByteString data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom(byte[] data) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + byte[] data, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return PARSER.parseFrom(data, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseDelimitedFrom(java.io.InputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseDelimitedFrom( + java.io.InputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseDelimitedWithIOException(PARSER, input, extensionRegistry); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + com.google.protobuf.CodedInputStream input) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input); + } + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parseFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + return com.google.protobuf.GeneratedMessageV3 + .parseWithIOException(PARSER, input, extensionRegistry); + } + + @java.lang.Override + public Builder newBuilderForType() { return newBuilder(); } + public static Builder newBuilder() { + return DEFAULT_INSTANCE.toBuilder(); + } + public static Builder newBuilder(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob prototype) { + return DEFAULT_INSTANCE.toBuilder().mergeFrom(prototype); + } + @java.lang.Override + public Builder toBuilder() { + return this == DEFAULT_INSTANCE + ? new Builder() : new Builder().mergeFrom(this); + } + + @java.lang.Override + protected Builder newBuilderForType( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + Builder builder = new Builder(parent); + return builder; + } + /** + *
+     * The spec of a training job. This is mostly a pass-through object
+     * https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html
+     * 
+ * + * Protobuf type {@code flyteidl.plugins.sagemaker.TrainingJob} + */ + public static final class Builder extends + com.google.protobuf.GeneratedMessageV3.Builder implements + // @@protoc_insertion_point(builder_implements:flyteidl.plugins.sagemaker.TrainingJob) + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobOrBuilder { + public static final com.google.protobuf.Descriptors.Descriptor + getDescriptor() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor; + } + + @java.lang.Override + protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internalGetFieldAccessorTable() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJob_fieldAccessorTable + .ensureFieldAccessorsInitialized( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.class, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.Builder.class); + } + + // Construct using flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.newBuilder() + private Builder() { + maybeForceBuilderInitialization(); + } + + private Builder( + com.google.protobuf.GeneratedMessageV3.BuilderParent parent) { + super(parent); + maybeForceBuilderInitialization(); + } + private void maybeForceBuilderInitialization() { + if (com.google.protobuf.GeneratedMessageV3 + .alwaysUseFieldBuilders) { + } + } + @java.lang.Override + public Builder clear() { + super.clear(); + if (algorithmSpecificationBuilder_ == null) { + algorithmSpecification_ = null; + } else { + algorithmSpecification_ = null; + algorithmSpecificationBuilder_ = null; + } + if (trainingJobResourceConfigBuilder_ == null) { + trainingJobResourceConfig_ = null; + } else { + trainingJobResourceConfig_ = null; + trainingJobResourceConfigBuilder_ = null; + } + return this; + } + + @java.lang.Override + public com.google.protobuf.Descriptors.Descriptor + getDescriptorForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getDefaultInstanceForType() { + return flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.getDefaultInstance(); + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob build() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob result = buildPartial(); + if (!result.isInitialized()) { + throw newUninitializedMessageException(result); + } + return result; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob buildPartial() { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob result = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob(this); + if (algorithmSpecificationBuilder_ == null) { + result.algorithmSpecification_ = algorithmSpecification_; + } else { + result.algorithmSpecification_ = algorithmSpecificationBuilder_.build(); + } + if (trainingJobResourceConfigBuilder_ == null) { + result.trainingJobResourceConfig_ = trainingJobResourceConfig_; + } else { + result.trainingJobResourceConfig_ = trainingJobResourceConfigBuilder_.build(); + } + onBuilt(); + return result; + } + + @java.lang.Override + public Builder clone() { + return super.clone(); + } + @java.lang.Override + public Builder setField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.setField(field, value); + } + @java.lang.Override + public Builder clearField( + com.google.protobuf.Descriptors.FieldDescriptor field) { + return super.clearField(field); + } + @java.lang.Override + public Builder clearOneof( + com.google.protobuf.Descriptors.OneofDescriptor oneof) { + return super.clearOneof(oneof); + } + @java.lang.Override + public Builder setRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + int index, java.lang.Object value) { + return super.setRepeatedField(field, index, value); + } + @java.lang.Override + public Builder addRepeatedField( + com.google.protobuf.Descriptors.FieldDescriptor field, + java.lang.Object value) { + return super.addRepeatedField(field, value); + } + @java.lang.Override + public Builder mergeFrom(com.google.protobuf.Message other) { + if (other instanceof flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob) { + return mergeFrom((flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob)other); + } else { + super.mergeFrom(other); + return this; + } + } + + public Builder mergeFrom(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob other) { + if (other == flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob.getDefaultInstance()) return this; + if (other.hasAlgorithmSpecification()) { + mergeAlgorithmSpecification(other.getAlgorithmSpecification()); + } + if (other.hasTrainingJobResourceConfig()) { + mergeTrainingJobResourceConfig(other.getTrainingJobResourceConfig()); + } + this.mergeUnknownFields(other.unknownFields); + onChanged(); + return this; + } + + @java.lang.Override + public final boolean isInitialized() { + return true; + } + + @java.lang.Override + public Builder mergeFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws java.io.IOException { + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob parsedMessage = null; + try { + parsedMessage = PARSER.parsePartialFrom(input, extensionRegistry); + } catch (com.google.protobuf.InvalidProtocolBufferException e) { + parsedMessage = (flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob) e.getUnfinishedMessage(); + throw e.unwrapIOException(); + } finally { + if (parsedMessage != null) { + mergeFrom(parsedMessage); + } + } + return this; + } + + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification algorithmSpecification_; + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder> algorithmSpecificationBuilder_; + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public boolean hasAlgorithmSpecification() { + return algorithmSpecificationBuilder_ != null || algorithmSpecification_ != null; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification getAlgorithmSpecification() { + if (algorithmSpecificationBuilder_ == null) { + return algorithmSpecification_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.getDefaultInstance() : algorithmSpecification_; + } else { + return algorithmSpecificationBuilder_.getMessage(); + } + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public Builder setAlgorithmSpecification(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification value) { + if (algorithmSpecificationBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + algorithmSpecification_ = value; + onChanged(); + } else { + algorithmSpecificationBuilder_.setMessage(value); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public Builder setAlgorithmSpecification( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder builderForValue) { + if (algorithmSpecificationBuilder_ == null) { + algorithmSpecification_ = builderForValue.build(); + onChanged(); + } else { + algorithmSpecificationBuilder_.setMessage(builderForValue.build()); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public Builder mergeAlgorithmSpecification(flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification value) { + if (algorithmSpecificationBuilder_ == null) { + if (algorithmSpecification_ != null) { + algorithmSpecification_ = + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.newBuilder(algorithmSpecification_).mergeFrom(value).buildPartial(); + } else { + algorithmSpecification_ = value; + } + onChanged(); + } else { + algorithmSpecificationBuilder_.mergeFrom(value); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public Builder clearAlgorithmSpecification() { + if (algorithmSpecificationBuilder_ == null) { + algorithmSpecification_ = null; + onChanged(); + } else { + algorithmSpecification_ = null; + algorithmSpecificationBuilder_ = null; + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder getAlgorithmSpecificationBuilder() { + + onChanged(); + return getAlgorithmSpecificationFieldBuilder().getBuilder(); + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder getAlgorithmSpecificationOrBuilder() { + if (algorithmSpecificationBuilder_ != null) { + return algorithmSpecificationBuilder_.getMessageOrBuilder(); + } else { + return algorithmSpecification_ == null ? + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.getDefaultInstance() : algorithmSpecification_; + } + } + /** + * .flyteidl.plugins.sagemaker.AlgorithmSpecification algorithm_specification = 1; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder> + getAlgorithmSpecificationFieldBuilder() { + if (algorithmSpecificationBuilder_ == null) { + algorithmSpecificationBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecification.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.AlgorithmSpecificationOrBuilder>( + getAlgorithmSpecification(), + getParentForChildren(), + isClean()); + algorithmSpecification_ = null; + } + return algorithmSpecificationBuilder_; + } + + private flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig trainingJobResourceConfig_; + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder> trainingJobResourceConfigBuilder_; + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public boolean hasTrainingJobResourceConfig() { + return trainingJobResourceConfigBuilder_ != null || trainingJobResourceConfig_ != null; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig getTrainingJobResourceConfig() { + if (trainingJobResourceConfigBuilder_ == null) { + return trainingJobResourceConfig_ == null ? flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.getDefaultInstance() : trainingJobResourceConfig_; + } else { + return trainingJobResourceConfigBuilder_.getMessage(); + } + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public Builder setTrainingJobResourceConfig(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig value) { + if (trainingJobResourceConfigBuilder_ == null) { + if (value == null) { + throw new NullPointerException(); + } + trainingJobResourceConfig_ = value; + onChanged(); + } else { + trainingJobResourceConfigBuilder_.setMessage(value); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public Builder setTrainingJobResourceConfig( + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder builderForValue) { + if (trainingJobResourceConfigBuilder_ == null) { + trainingJobResourceConfig_ = builderForValue.build(); + onChanged(); + } else { + trainingJobResourceConfigBuilder_.setMessage(builderForValue.build()); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public Builder mergeTrainingJobResourceConfig(flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig value) { + if (trainingJobResourceConfigBuilder_ == null) { + if (trainingJobResourceConfig_ != null) { + trainingJobResourceConfig_ = + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.newBuilder(trainingJobResourceConfig_).mergeFrom(value).buildPartial(); + } else { + trainingJobResourceConfig_ = value; + } + onChanged(); + } else { + trainingJobResourceConfigBuilder_.mergeFrom(value); + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public Builder clearTrainingJobResourceConfig() { + if (trainingJobResourceConfigBuilder_ == null) { + trainingJobResourceConfig_ = null; + onChanged(); + } else { + trainingJobResourceConfig_ = null; + trainingJobResourceConfigBuilder_ = null; + } + + return this; + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder getTrainingJobResourceConfigBuilder() { + + onChanged(); + return getTrainingJobResourceConfigFieldBuilder().getBuilder(); + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder getTrainingJobResourceConfigOrBuilder() { + if (trainingJobResourceConfigBuilder_ != null) { + return trainingJobResourceConfigBuilder_.getMessageOrBuilder(); + } else { + return trainingJobResourceConfig_ == null ? + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.getDefaultInstance() : trainingJobResourceConfig_; + } + } + /** + * .flyteidl.plugins.sagemaker.TrainingJobResourceConfig training_job_resource_config = 2; + */ + private com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder> + getTrainingJobResourceConfigFieldBuilder() { + if (trainingJobResourceConfigBuilder_ == null) { + trainingJobResourceConfigBuilder_ = new com.google.protobuf.SingleFieldBuilderV3< + flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfig.Builder, flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJobResourceConfigOrBuilder>( + getTrainingJobResourceConfig(), + getParentForChildren(), + isClean()); + trainingJobResourceConfig_ = null; + } + return trainingJobResourceConfigBuilder_; + } + @java.lang.Override + public final Builder setUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.setUnknownFields(unknownFields); + } + + @java.lang.Override + public final Builder mergeUnknownFields( + final com.google.protobuf.UnknownFieldSet unknownFields) { + return super.mergeUnknownFields(unknownFields); + } + + + // @@protoc_insertion_point(builder_scope:flyteidl.plugins.sagemaker.TrainingJob) + } + + // @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJob) + private static final flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob DEFAULT_INSTANCE; + static { + DEFAULT_INSTANCE = new flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob(); + } + + public static flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getDefaultInstance() { + return DEFAULT_INSTANCE; + } + + private static final com.google.protobuf.Parser + PARSER = new com.google.protobuf.AbstractParser() { + @java.lang.Override + public TrainingJob parsePartialFrom( + com.google.protobuf.CodedInputStream input, + com.google.protobuf.ExtensionRegistryLite extensionRegistry) + throws com.google.protobuf.InvalidProtocolBufferException { + return new TrainingJob(input, extensionRegistry); + } + }; + + public static com.google.protobuf.Parser parser() { + return PARSER; + } + + @java.lang.Override + public com.google.protobuf.Parser getParserForType() { + return PARSER; + } + + @java.lang.Override + public flyteidl.plugins.sagemaker.TrainingJobOuterClass.TrainingJob getDefaultInstanceForType() { + return DEFAULT_INSTANCE; + } + + } + + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_InputMode_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_AlgorithmName_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_InputContentType_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_MetricDefinition_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_fieldAccessorTable; + private static final com.google.protobuf.Descriptors.Descriptor + internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor; + private static final + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable + internal_static_flyteidl_plugins_sagemaker_TrainingJob_fieldAccessorTable; + + public static com.google.protobuf.Descriptors.FileDescriptor + getDescriptor() { + return descriptor; + } + private static com.google.protobuf.Descriptors.FileDescriptor + descriptor; + static { + java.lang.String[] descriptorData = { + "\n-flyteidl/plugins/sagemaker/training_jo" + + "b.proto\022\032flyteidl.plugins.sagemaker\032\036goo" + + "gle/protobuf/duration.proto\"(\n\tInputMode" + + "\"\033\n\005Value\022\010\n\004FILE\020\000\022\010\n\004PIPE\020\001\"1\n\rAlgorit" + + "hmName\" \n\005Value\022\n\n\006CUSTOM\020\000\022\013\n\007XGBOOST\020\001" + + "\")\n\020InputContentType\"\025\n\005Value\022\014\n\010TEXT_CS" + + "V\020\000\"/\n\020MetricDefinition\022\014\n\004name\030\001 \001(\t\022\r\n" + + "\005regex\030\002 \001(\t\"\327\002\n\026AlgorithmSpecification\022" + + "?\n\ninput_mode\030\001 \001(\0162+.flyteidl.plugins.s" + + "agemaker.InputMode.Value\022G\n\016algorithm_na" + + "me\030\002 \001(\0162/.flyteidl.plugins.sagemaker.Al" + + "gorithmName.Value\022\031\n\021algorithm_version\030\003" + + " \001(\t\022H\n\022metric_definitions\030\004 \003(\0132,.flyte" + + "idl.plugins.sagemaker.MetricDefinition\022N" + + "\n\022input_content_type\030\005 \001(\01622.flyteidl.pl" + + "ugins.sagemaker.InputContentType.Value\"e" + + "\n\031TrainingJobResourceConfig\022\026\n\016instance_" + + "count\030\001 \001(\003\022\025\n\rinstance_type\030\002 \001(\t\022\031\n\021vo" + + "lume_size_in_gb\030\003 \001(\003\"\277\001\n\013TrainingJob\022S\n" + + "\027algorithm_specification\030\001 \001(\01322.flyteid" + + "l.plugins.sagemaker.AlgorithmSpecificati" + + "on\022[\n\034training_job_resource_config\030\002 \001(\013" + + "25.flyteidl.plugins.sagemaker.TrainingJo" + + "bResourceConfigB5Z3github.com/lyft/flyte" + + "idl/gen/pb-go/flyteidl/pluginsb\006proto3" + }; + com.google.protobuf.Descriptors.FileDescriptor.InternalDescriptorAssigner assigner = + new com.google.protobuf.Descriptors.FileDescriptor. InternalDescriptorAssigner() { + public com.google.protobuf.ExtensionRegistry assignDescriptors( + com.google.protobuf.Descriptors.FileDescriptor root) { + descriptor = root; + return null; + } + }; + com.google.protobuf.Descriptors.FileDescriptor + .internalBuildGeneratedFileFrom(descriptorData, + new com.google.protobuf.Descriptors.FileDescriptor[] { + com.google.protobuf.DurationProto.getDescriptor(), + }, assigner); + internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor = + getDescriptor().getMessageTypes().get(0); + internal_static_flyteidl_plugins_sagemaker_InputMode_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_InputMode_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor = + getDescriptor().getMessageTypes().get(1); + internal_static_flyteidl_plugins_sagemaker_AlgorithmName_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_AlgorithmName_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor = + getDescriptor().getMessageTypes().get(2); + internal_static_flyteidl_plugins_sagemaker_InputContentType_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_InputContentType_descriptor, + new java.lang.String[] { }); + internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor = + getDescriptor().getMessageTypes().get(3); + internal_static_flyteidl_plugins_sagemaker_MetricDefinition_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_MetricDefinition_descriptor, + new java.lang.String[] { "Name", "Regex", }); + internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor = + getDescriptor().getMessageTypes().get(4); + internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_AlgorithmSpecification_descriptor, + new java.lang.String[] { "InputMode", "AlgorithmName", "AlgorithmVersion", "MetricDefinitions", "InputContentType", }); + internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor = + getDescriptor().getMessageTypes().get(5); + internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_TrainingJobResourceConfig_descriptor, + new java.lang.String[] { "InstanceCount", "InstanceType", "VolumeSizeInGb", }); + internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor = + getDescriptor().getMessageTypes().get(6); + internal_static_flyteidl_plugins_sagemaker_TrainingJob_fieldAccessorTable = new + com.google.protobuf.GeneratedMessageV3.FieldAccessorTable( + internal_static_flyteidl_plugins_sagemaker_TrainingJob_descriptor, + new java.lang.String[] { "AlgorithmSpecification", "TrainingJobResourceConfig", }); + com.google.protobuf.DurationProto.getDescriptor(); + } + + // @@protoc_insertion_point(outer_class_scope) +} diff --git a/gen/pb-protodoc/flyteidl/plugins/index.rst b/gen/pb-protodoc/flyteidl/plugins/index.rst index a44c59d49..7cb2db6f7 100644 --- a/gen/pb-protodoc/flyteidl/plugins/index.rst +++ b/gen/pb-protodoc/flyteidl/plugins/index.rst @@ -10,6 +10,7 @@ Plugins available in the Flyte system. :caption: plugins :name: pluginstoc + sagemaker/index array_job.proto presto.proto pytorch.proto diff --git a/gen/pb-protodoc/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto.rst b/gen/pb-protodoc/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto.rst new file mode 100644 index 000000000..59fe2eb72 --- /dev/null +++ b/gen/pb-protodoc/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto.rst @@ -0,0 +1,230 @@ +.. _api_file_flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto: + +hyperparameter_tuning_job.proto +========================================================== + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterTuningJob: + +flyteidl.plugins.sagemaker.HyperparameterTuningJob +-------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterTuningJob proto] `_ + +A pass-through for SageMaker's hyperparameter tuning job + +.. code-block:: json + + { + "training_job": "{...}", + "max_number_of_training_jobs": "...", + "max_parallel_training_jobs": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job: + +training_job + (:ref:`flyteidl.plugins.sagemaker.TrainingJob `) The underlying training job that the hyperparameter tuning job will launch during the process + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_number_of_training_jobs: + +max_number_of_training_jobs + (`int64 `_) The maximum number of training jobs that an hpo job can launch. For resource limit purpose. + https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_parallel_training_jobs: + +max_parallel_training_jobs + (`int64 `_) The maximum number of concurrent training job that an hpo job can launch + https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + + + + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType: + +flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType +------------------------------------------------------------ + +`[flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType proto] `_ + +HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job +with respect to the specified metric. + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value: + +Enum flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value +----------------------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value.MINIMIZE: + +MINIMIZE + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value.MAXIMIZE: + +MAXIMIZE + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterTuningObjective: + +flyteidl.plugins.sagemaker.HyperparameterTuningObjective +-------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterTuningObjective proto] `_ + +The target metric and the objective of the hyperparameter tuning. +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html + +.. code-block:: json + + { + "objective_type": "...", + "metric_name": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningObjective.objective_type: + +objective_type + (:ref:`flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value `) HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job + with respect to the specified metric. + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name: + +metric_name + (`string `_) The target metric name, which is the user-defined name of the metric specified in the + training job's algorithm specification + + + + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterTuningStrategy: + +flyteidl.plugins.sagemaker.HyperparameterTuningStrategy +------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterTuningStrategy proto] `_ + +Setting the strategy used when searching in the hyperparameter space +Refer this doc for more details: +https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value: + +Enum flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value +------------------------------------------------------------------ + +`[flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value.BAYESIAN: + +BAYESIAN + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value.RANDOM: + +RANDOM + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType: + +flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType +------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType proto] `_ + +When the training jobs launched by the hyperparameter tuning job are not improving significantly, +a hyperparameter tuning job can be stopping early. +Note that there's only a subset of built-in algorithms that supports early stopping. +see: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-early-stopping.html + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value: + +Enum flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value +------------------------------------------------------------------ + +`[flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value.OFF: + +OFF + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value.AUTO: + +AUTO + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig: + +flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig +-------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig proto] `_ + +The specification of the hyperparameter tuning process +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config + +.. code-block:: json + + { + "hyperparameter_ranges": "{...}", + "tuning_strategy": "...", + "tuning_objective": "{...}", + "training_job_early_stopping_type": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges: + +hyperparameter_ranges + (:ref:`flyteidl.plugins.sagemaker.ParameterRanges `) ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_strategy: + +tuning_strategy + (:ref:`flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value `) Setting the strategy used when searching in the hyperparameter space + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective: + +tuning_objective + (:ref:`flyteidl.plugins.sagemaker.HyperparameterTuningObjective `) The target metric and the objective of the hyperparameter tuning. + + +.. _api_field_flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.training_job_early_stopping_type: + +training_job_early_stopping_type + (:ref:`flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value `) When the training jobs launched by the hyperparameter tuning job are not improving significantly, + a hyperparameter tuning job can be stopping early. + + + diff --git a/gen/pb-protodoc/flyteidl/plugins/sagemaker/index.rst b/gen/pb-protodoc/flyteidl/plugins/sagemaker/index.rst new file mode 100644 index 000000000..667b8213d --- /dev/null +++ b/gen/pb-protodoc/flyteidl/plugins/sagemaker/index.rst @@ -0,0 +1,11 @@ +sagemaker +========= + +.. toctree:: + :maxdepth: 1 + :caption: sagemaker + :name: sagemakertoc + + hyperparameter_tuning_job.proto + parameter_ranges.proto + training_job.proto diff --git a/gen/pb-protodoc/flyteidl/plugins/sagemaker/parameter_ranges.proto.rst b/gen/pb-protodoc/flyteidl/plugins/sagemaker/parameter_ranges.proto.rst new file mode 100644 index 000000000..ab5e14097 --- /dev/null +++ b/gen/pb-protodoc/flyteidl/plugins/sagemaker/parameter_ranges.proto.rst @@ -0,0 +1,215 @@ +.. _api_file_flyteidl/plugins/sagemaker/parameter_ranges.proto: + +parameter_ranges.proto +================================================= + +.. _api_msg_flyteidl.plugins.sagemaker.HyperparameterScalingType: + +flyteidl.plugins.sagemaker.HyperparameterScalingType +---------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterScalingType proto] `_ + +HyperparameterScalingType defines the way to increase or decrease the value of the hyperparameter +For details, refer to: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +See examples of these scaling type, refer to: https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.HyperparameterScalingType.Value: + +Enum flyteidl.plugins.sagemaker.HyperparameterScalingType.Value +--------------------------------------------------------------- + +`[flyteidl.plugins.sagemaker.HyperparameterScalingType.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterScalingType.Value.AUTO: + +AUTO + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterScalingType.Value.LINEAR: + +LINEAR + ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterScalingType.Value.LOGARITHMIC: + +LOGARITHMIC + ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.HyperparameterScalingType.Value.REVERSELOGARITHMIC: + +REVERSELOGARITHMIC + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.ContinuousParameterRange: + +flyteidl.plugins.sagemaker.ContinuousParameterRange +--------------------------------------------------- + +`[flyteidl.plugins.sagemaker.ContinuousParameterRange proto] `_ + +ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +users to specify the search space of a floating-point hyperparameter +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html + +.. code-block:: json + + { + "max_value": "...", + "min_value": "...", + "scaling_type": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.ContinuousParameterRange.max_value: + +max_value + (`double `_) + +.. _api_field_flyteidl.plugins.sagemaker.ContinuousParameterRange.min_value: + +min_value + (`double `_) + +.. _api_field_flyteidl.plugins.sagemaker.ContinuousParameterRange.scaling_type: + +scaling_type + (:ref:`flyteidl.plugins.sagemaker.HyperparameterScalingType.Value `) + + + +.. _api_msg_flyteidl.plugins.sagemaker.IntegerParameterRange: + +flyteidl.plugins.sagemaker.IntegerParameterRange +------------------------------------------------ + +`[flyteidl.plugins.sagemaker.IntegerParameterRange proto] `_ + +IntegerParameterRange refers to a discrete range of hyperparameter values, allowing +users to specify the search space of an integer hyperparameter +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html + +.. code-block:: json + + { + "max_value": "...", + "min_value": "...", + "scaling_type": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.IntegerParameterRange.max_value: + +max_value + (`int64 `_) + +.. _api_field_flyteidl.plugins.sagemaker.IntegerParameterRange.min_value: + +min_value + (`int64 `_) + +.. _api_field_flyteidl.plugins.sagemaker.IntegerParameterRange.scaling_type: + +scaling_type + (:ref:`flyteidl.plugins.sagemaker.HyperparameterScalingType.Value `) + + + +.. _api_msg_flyteidl.plugins.sagemaker.CategoricalParameterRange: + +flyteidl.plugins.sagemaker.CategoricalParameterRange +---------------------------------------------------- + +`[flyteidl.plugins.sagemaker.CategoricalParameterRange proto] `_ + +ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +users to specify the search space of a floating-point hyperparameter +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html + +.. code-block:: json + + { + "values": [] + } + +.. _api_field_flyteidl.plugins.sagemaker.CategoricalParameterRange.values: + +values + (`string `_) + + + +.. _api_msg_flyteidl.plugins.sagemaker.ParameterRangeOneOf: + +flyteidl.plugins.sagemaker.ParameterRangeOneOf +---------------------------------------------- + +`[flyteidl.plugins.sagemaker.ParameterRangeOneOf proto] `_ + +ParameterRangeOneOf describes a single ParameterRange, which is a one-of structure that can be one of +the three possible types: ContinuousParameterRange, IntegerParameterRange, and CategoricalParameterRange. +This one-of structure in Flyte enables specifying a Parameter in a type-safe manner +See: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html + +.. code-block:: json + + { + "continuous_parameter_range": "{...}", + "integer_parameter_range": "{...}", + "categorical_parameter_range": "{...}" + } + +.. _api_field_flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range: + +continuous_parameter_range + (:ref:`flyteidl.plugins.sagemaker.ContinuousParameterRange `) + + + Only one of :ref:`continuous_parameter_range `, :ref:`integer_parameter_range `, :ref:`categorical_parameter_range ` may be set. + +.. _api_field_flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range: + +integer_parameter_range + (:ref:`flyteidl.plugins.sagemaker.IntegerParameterRange `) + + + Only one of :ref:`continuous_parameter_range `, :ref:`integer_parameter_range `, :ref:`categorical_parameter_range ` may be set. + +.. _api_field_flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range: + +categorical_parameter_range + (:ref:`flyteidl.plugins.sagemaker.CategoricalParameterRange `) + + + Only one of :ref:`continuous_parameter_range `, :ref:`integer_parameter_range `, :ref:`categorical_parameter_range ` may be set. + + + +.. _api_msg_flyteidl.plugins.sagemaker.ParameterRanges: + +flyteidl.plugins.sagemaker.ParameterRanges +------------------------------------------ + +`[flyteidl.plugins.sagemaker.ParameterRanges proto] `_ + +ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range +https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html + +.. code-block:: json + + { + "parameter_range_map": "{...}" + } + +.. _api_field_flyteidl.plugins.sagemaker.ParameterRanges.parameter_range_map: + +parameter_range_map + (map<`string `_, :ref:`flyteidl.plugins.sagemaker.ParameterRangeOneOf `>) + + diff --git a/gen/pb-protodoc/flyteidl/plugins/sagemaker/training_job.proto.rst b/gen/pb-protodoc/flyteidl/plugins/sagemaker/training_job.proto.rst new file mode 100644 index 000000000..dda63465b --- /dev/null +++ b/gen/pb-protodoc/flyteidl/plugins/sagemaker/training_job.proto.rst @@ -0,0 +1,274 @@ +.. _api_file_flyteidl/plugins/sagemaker/training_job.proto: + +training_job.proto +============================================= + +.. _api_msg_flyteidl.plugins.sagemaker.InputMode: + +flyteidl.plugins.sagemaker.InputMode +------------------------------------ + +`[flyteidl.plugins.sagemaker.InputMode proto] `_ + +The input mode that the algorithm supports. When using the File input mode, SageMaker downloads +the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker +volume for training container. When using Pipe input mode, Amazon SageMaker streams data directly +from S3 to the container. +See: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html +For the input modes that different SageMaker algorithms support, see: +https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.InputMode.Value: + +Enum flyteidl.plugins.sagemaker.InputMode.Value +----------------------------------------------- + +`[flyteidl.plugins.sagemaker.InputMode.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.InputMode.Value.FILE: + +FILE + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.InputMode.Value.PIPE: + +PIPE + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.AlgorithmName: + +flyteidl.plugins.sagemaker.AlgorithmName +---------------------------------------- + +`[flyteidl.plugins.sagemaker.AlgorithmName proto] `_ + +The algorithm name is used for deciding which pre-built image to point to. +This is only required for use cases where SageMaker's built-in algorithm mode is used. +While we currently only support a subset of the algorithms, more will be added to the list. +See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.AlgorithmName.Value: + +Enum flyteidl.plugins.sagemaker.AlgorithmName.Value +--------------------------------------------------- + +`[flyteidl.plugins.sagemaker.AlgorithmName.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.AlgorithmName.Value.CUSTOM: + +CUSTOM + *(DEFAULT)* ⁣ + +.. _api_enum_value_flyteidl.plugins.sagemaker.AlgorithmName.Value.XGBOOST: + +XGBOOST + ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.InputContentType: + +flyteidl.plugins.sagemaker.InputContentType +------------------------------------------- + +`[flyteidl.plugins.sagemaker.InputContentType proto] `_ + +Specifies the type of file for input data. Different SageMaker built-in algorithms require different file types of input data +See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html +https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html + +.. code-block:: json + + {} + + + +.. _api_enum_flyteidl.plugins.sagemaker.InputContentType.Value: + +Enum flyteidl.plugins.sagemaker.InputContentType.Value +------------------------------------------------------ + +`[flyteidl.plugins.sagemaker.InputContentType.Value proto] `_ + + +.. _api_enum_value_flyteidl.plugins.sagemaker.InputContentType.Value.TEXT_CSV: + +TEXT_CSV + *(DEFAULT)* ⁣ + + +.. _api_msg_flyteidl.plugins.sagemaker.MetricDefinition: + +flyteidl.plugins.sagemaker.MetricDefinition +------------------------------------------- + +`[flyteidl.plugins.sagemaker.MetricDefinition proto] `_ + +Specifies a metric that the training algorithm writes to stderr or stdout. +This object is a pass-through. +See this for details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MetricDefinition.html + +.. code-block:: json + + { + "name": "...", + "regex": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.MetricDefinition.name: + +name + (`string `_) User-defined name of the metric + + +.. _api_field_flyteidl.plugins.sagemaker.MetricDefinition.regex: + +regex + (`string `_) SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics + + + + +.. _api_msg_flyteidl.plugins.sagemaker.AlgorithmSpecification: + +flyteidl.plugins.sagemaker.AlgorithmSpecification +------------------------------------------------- + +`[flyteidl.plugins.sagemaker.AlgorithmSpecification proto] `_ + +Specifies the training algorithm to be used in the training job +This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify +TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm +name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to +CUSTOM. In this case, the value of the algorithm_version field has no effect +For pass-through use cases: refer to this AWS official document for more details +https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html + +.. code-block:: json + + { + "input_mode": "...", + "algorithm_name": "...", + "algorithm_version": "...", + "metric_definitions": [], + "input_content_type": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.AlgorithmSpecification.input_mode: + +input_mode + (:ref:`flyteidl.plugins.sagemaker.InputMode.Value `) The input mode can be either PIPE or FILE + + +.. _api_field_flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_name: + +algorithm_name + (:ref:`flyteidl.plugins.sagemaker.AlgorithmName.Value `) The algorithm name is used for deciding which pre-built image to point to + + +.. _api_field_flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version: + +algorithm_version + (`string `_) The algorithm version field is used for deciding which pre-built image to point to + This is only needed for use cases where SageMaker's built-in algorithm mode is chosen + + +.. _api_field_flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions: + +metric_definitions + (:ref:`flyteidl.plugins.sagemaker.MetricDefinition `) A list of metric definitions for SageMaker to evaluate/track on the progress of the training job + See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html + and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html + + +.. _api_field_flyteidl.plugins.sagemaker.AlgorithmSpecification.input_content_type: + +input_content_type + (:ref:`flyteidl.plugins.sagemaker.InputContentType.Value `) The content type of the input + See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html + https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html + + + + +.. _api_msg_flyteidl.plugins.sagemaker.TrainingJobResourceConfig: + +flyteidl.plugins.sagemaker.TrainingJobResourceConfig +---------------------------------------------------- + +`[flyteidl.plugins.sagemaker.TrainingJobResourceConfig proto] `_ + +TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the +number of instances to launch, and the size of the ML storage volume the user wants to provision +Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html + +.. code-block:: json + + { + "instance_count": "...", + "instance_type": "...", + "volume_size_in_gb": "..." + } + +.. _api_field_flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_count: + +instance_count + (`int64 `_) The number of ML compute instances to use. For distributed training, provide a value greater than 1. + + +.. _api_field_flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type: + +instance_type + (`string `_) The ML compute instance type + + +.. _api_field_flyteidl.plugins.sagemaker.TrainingJobResourceConfig.volume_size_in_gb: + +volume_size_in_gb + (`int64 `_) The size of the ML storage volume that you want to provision. + + + + +.. _api_msg_flyteidl.plugins.sagemaker.TrainingJob: + +flyteidl.plugins.sagemaker.TrainingJob +-------------------------------------- + +`[flyteidl.plugins.sagemaker.TrainingJob proto] `_ + +The spec of a training job. This is mostly a pass-through object +https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html + +.. code-block:: json + + { + "algorithm_specification": "{...}", + "training_job_resource_config": "{...}" + } + +.. _api_field_flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification: + +algorithm_specification + (:ref:`flyteidl.plugins.sagemaker.AlgorithmSpecification `) + +.. _api_field_flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config: + +training_job_resource_config + (:ref:`flyteidl.plugins.sagemaker.TrainingJobResourceConfig `) + + diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/__init__.py b/gen/pb_python/flyteidl/plugins/sagemaker/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2.py b/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2.py new file mode 100644 index 000000000..68c53659e --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2.py @@ -0,0 +1,368 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from flyteidl.plugins.sagemaker import parameter_ranges_pb2 as flyteidl_dot_plugins_dot_sagemaker_dot_parameter__ranges__pb2 +from flyteidl.plugins.sagemaker import training_job_pb2 as flyteidl_dot_plugins_dot_sagemaker_dot_training__job__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto', + package='flyteidl.plugins.sagemaker', + syntax='proto3', + serialized_options=_b('Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins'), + serialized_pb=_b('\n:flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto\x12\x1a\x66lyteidl.plugins.sagemaker\x1a\x31\x66lyteidl/plugins/sagemaker/parameter_ranges.proto\x1a-flyteidl/plugins/sagemaker/training_job.proto\"\xa1\x01\n\x17HyperparameterTuningJob\x12=\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\'.flyteidl.plugins.sagemaker.TrainingJob\x12#\n\x1bmax_number_of_training_jobs\x18\x02 \x01(\x03\x12\"\n\x1amax_parallel_training_jobs\x18\x03 \x01(\x03\"H\n!HyperparameterTuningObjectiveType\"#\n\x05Value\x12\x0c\n\x08MINIMIZE\x10\x00\x12\x0c\n\x08MAXIMIZE\x10\x01\"\x91\x01\n\x1dHyperparameterTuningObjective\x12[\n\x0eobjective_type\x18\x01 \x01(\x0e\x32\x43.flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value\x12\x13\n\x0bmetric_name\x18\x02 \x01(\t\"A\n\x1cHyperparameterTuningStrategy\"!\n\x05Value\x12\x0c\n\x08\x42\x41YESIAN\x10\x00\x12\n\n\x06RANDOM\x10\x01\":\n\x1cTrainingJobEarlyStoppingType\"\x1a\n\x05Value\x12\x07\n\x03OFF\x10\x00\x12\x08\n\x04\x41UTO\x10\x01\"\x83\x03\n\x1dHyperparameterTuningJobConfig\x12J\n\x15hyperparameter_ranges\x18\x01 \x01(\x0b\x32+.flyteidl.plugins.sagemaker.ParameterRanges\x12W\n\x0ftuning_strategy\x18\x02 \x01(\x0e\x32>.flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value\x12S\n\x10tuning_objective\x18\x03 \x01(\x0b\x32\x39.flyteidl.plugins.sagemaker.HyperparameterTuningObjective\x12h\n training_job_early_stopping_type\x18\x04 \x01(\x0e\x32>.flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.ValueB5Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/pluginsb\x06proto3') + , + dependencies=[flyteidl_dot_plugins_dot_sagemaker_dot_parameter__ranges__pb2.DESCRIPTOR,flyteidl_dot_plugins_dot_sagemaker_dot_training__job__pb2.DESCRIPTOR,]) + + + +_HYPERPARAMETERTUNINGOBJECTIVETYPE_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='MINIMIZE', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='MAXIMIZE', index=1, number=1, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=389, + serialized_end=424, +) +_sym_db.RegisterEnumDescriptor(_HYPERPARAMETERTUNINGOBJECTIVETYPE_VALUE) + +_HYPERPARAMETERTUNINGSTRATEGY_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningStrategy.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='BAYESIAN', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='RANDOM', index=1, number=1, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=606, + serialized_end=639, +) +_sym_db.RegisterEnumDescriptor(_HYPERPARAMETERTUNINGSTRATEGY_VALUE) + +_TRAININGJOBEARLYSTOPPINGTYPE_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='OFF', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='AUTO', index=1, number=1, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=673, + serialized_end=699, +) +_sym_db.RegisterEnumDescriptor(_TRAININGJOBEARLYSTOPPINGTYPE_VALUE) + + +_HYPERPARAMETERTUNINGJOB = _descriptor.Descriptor( + name='HyperparameterTuningJob', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJob', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='training_job', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJob.training_job', index=0, + number=1, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_number_of_training_jobs', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_number_of_training_jobs', index=1, + number=2, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='max_parallel_training_jobs', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJob.max_parallel_training_jobs', index=2, + number=3, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=189, + serialized_end=350, +) + + +_HYPERPARAMETERTUNINGOBJECTIVETYPE = _descriptor.Descriptor( + name='HyperparameterTuningObjectiveType', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _HYPERPARAMETERTUNINGOBJECTIVETYPE_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=352, + serialized_end=424, +) + + +_HYPERPARAMETERTUNINGOBJECTIVE = _descriptor.Descriptor( + name='HyperparameterTuningObjective', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningObjective', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='objective_type', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningObjective.objective_type', index=0, + number=1, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='metric_name', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningObjective.metric_name', index=1, + number=2, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=427, + serialized_end=572, +) + + +_HYPERPARAMETERTUNINGSTRATEGY = _descriptor.Descriptor( + name='HyperparameterTuningStrategy', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningStrategy', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _HYPERPARAMETERTUNINGSTRATEGY_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=574, + serialized_end=639, +) + + +_TRAININGJOBEARLYSTOPPINGTYPE = _descriptor.Descriptor( + name='TrainingJobEarlyStoppingType', + full_name='flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _TRAININGJOBEARLYSTOPPINGTYPE_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=641, + serialized_end=699, +) + + +_HYPERPARAMETERTUNINGJOBCONFIG = _descriptor.Descriptor( + name='HyperparameterTuningJobConfig', + full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='hyperparameter_ranges', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.hyperparameter_ranges', index=0, + number=1, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='tuning_strategy', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_strategy', index=1, + number=2, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='tuning_objective', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.tuning_objective', index=2, + number=3, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='training_job_early_stopping_type', full_name='flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig.training_job_early_stopping_type', index=3, + number=4, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=702, + serialized_end=1089, +) + +_HYPERPARAMETERTUNINGJOB.fields_by_name['training_job'].message_type = flyteidl_dot_plugins_dot_sagemaker_dot_training__job__pb2._TRAININGJOB +_HYPERPARAMETERTUNINGOBJECTIVETYPE_VALUE.containing_type = _HYPERPARAMETERTUNINGOBJECTIVETYPE +_HYPERPARAMETERTUNINGOBJECTIVE.fields_by_name['objective_type'].enum_type = _HYPERPARAMETERTUNINGOBJECTIVETYPE_VALUE +_HYPERPARAMETERTUNINGSTRATEGY_VALUE.containing_type = _HYPERPARAMETERTUNINGSTRATEGY +_TRAININGJOBEARLYSTOPPINGTYPE_VALUE.containing_type = _TRAININGJOBEARLYSTOPPINGTYPE +_HYPERPARAMETERTUNINGJOBCONFIG.fields_by_name['hyperparameter_ranges'].message_type = flyteidl_dot_plugins_dot_sagemaker_dot_parameter__ranges__pb2._PARAMETERRANGES +_HYPERPARAMETERTUNINGJOBCONFIG.fields_by_name['tuning_strategy'].enum_type = _HYPERPARAMETERTUNINGSTRATEGY_VALUE +_HYPERPARAMETERTUNINGJOBCONFIG.fields_by_name['tuning_objective'].message_type = _HYPERPARAMETERTUNINGOBJECTIVE +_HYPERPARAMETERTUNINGJOBCONFIG.fields_by_name['training_job_early_stopping_type'].enum_type = _TRAININGJOBEARLYSTOPPINGTYPE_VALUE +DESCRIPTOR.message_types_by_name['HyperparameterTuningJob'] = _HYPERPARAMETERTUNINGJOB +DESCRIPTOR.message_types_by_name['HyperparameterTuningObjectiveType'] = _HYPERPARAMETERTUNINGOBJECTIVETYPE +DESCRIPTOR.message_types_by_name['HyperparameterTuningObjective'] = _HYPERPARAMETERTUNINGOBJECTIVE +DESCRIPTOR.message_types_by_name['HyperparameterTuningStrategy'] = _HYPERPARAMETERTUNINGSTRATEGY +DESCRIPTOR.message_types_by_name['TrainingJobEarlyStoppingType'] = _TRAININGJOBEARLYSTOPPINGTYPE +DESCRIPTOR.message_types_by_name['HyperparameterTuningJobConfig'] = _HYPERPARAMETERTUNINGJOBCONFIG +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +HyperparameterTuningJob = _reflection.GeneratedProtocolMessageType('HyperparameterTuningJob', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERTUNINGJOB, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJob) + )) +_sym_db.RegisterMessage(HyperparameterTuningJob) + +HyperparameterTuningObjectiveType = _reflection.GeneratedProtocolMessageType('HyperparameterTuningObjectiveType', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERTUNINGOBJECTIVETYPE, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjectiveType) + )) +_sym_db.RegisterMessage(HyperparameterTuningObjectiveType) + +HyperparameterTuningObjective = _reflection.GeneratedProtocolMessageType('HyperparameterTuningObjective', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERTUNINGOBJECTIVE, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningObjective) + )) +_sym_db.RegisterMessage(HyperparameterTuningObjective) + +HyperparameterTuningStrategy = _reflection.GeneratedProtocolMessageType('HyperparameterTuningStrategy', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERTUNINGSTRATEGY, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningStrategy) + )) +_sym_db.RegisterMessage(HyperparameterTuningStrategy) + +TrainingJobEarlyStoppingType = _reflection.GeneratedProtocolMessageType('TrainingJobEarlyStoppingType', (_message.Message,), dict( + DESCRIPTOR = _TRAININGJOBEARLYSTOPPINGTYPE, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobEarlyStoppingType) + )) +_sym_db.RegisterMessage(TrainingJobEarlyStoppingType) + +HyperparameterTuningJobConfig = _reflection.GeneratedProtocolMessageType('HyperparameterTuningJobConfig', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERTUNINGJOBCONFIG, + __module__ = 'flyteidl.plugins.sagemaker.hyperparameter_tuning_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterTuningJobConfig) + )) +_sym_db.RegisterMessage(HyperparameterTuningJobConfig) + + +DESCRIPTOR._options = None +# @@protoc_insertion_point(module_scope) diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2_grpc.py b/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2_grpc.py new file mode 100644 index 000000000..a89435267 --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/hyperparameter_tuning_job_pb2_grpc.py @@ -0,0 +1,3 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +import grpc + diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2.py b/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2.py new file mode 100644 index 000000000..75f009d09 --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2.py @@ -0,0 +1,399 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: flyteidl/plugins/sagemaker/parameter_ranges.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='flyteidl/plugins/sagemaker/parameter_ranges.proto', + package='flyteidl.plugins.sagemaker', + syntax='proto3', + serialized_options=_b('Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins'), + serialized_pb=_b('\n1flyteidl/plugins/sagemaker/parameter_ranges.proto\x12\x1a\x66lyteidl.plugins.sagemaker\"c\n\x19HyperparameterScalingType\"F\n\x05Value\x12\x08\n\x04\x41UTO\x10\x00\x12\n\n\x06LINEAR\x10\x01\x12\x0f\n\x0bLOGARITHMIC\x10\x02\x12\x16\n\x12REVERSELOGARITHMIC\x10\x03\"\x93\x01\n\x18\x43ontinuousParameterRange\x12\x11\n\tmax_value\x18\x01 \x01(\x01\x12\x11\n\tmin_value\x18\x02 \x01(\x01\x12Q\n\x0cscaling_type\x18\x03 \x01(\x0e\x32;.flyteidl.plugins.sagemaker.HyperparameterScalingType.Value\"\x90\x01\n\x15IntegerParameterRange\x12\x11\n\tmax_value\x18\x01 \x01(\x03\x12\x11\n\tmin_value\x18\x02 \x01(\x03\x12Q\n\x0cscaling_type\x18\x03 \x01(\x0e\x32;.flyteidl.plugins.sagemaker.HyperparameterScalingType.Value\"+\n\x19\x43\x61tegoricalParameterRange\x12\x0e\n\x06values\x18\x01 \x03(\t\"\xbd\x02\n\x13ParameterRangeOneOf\x12Z\n\x1a\x63ontinuous_parameter_range\x18\x01 \x01(\x0b\x32\x34.flyteidl.plugins.sagemaker.ContinuousParameterRangeH\x00\x12T\n\x17integer_parameter_range\x18\x02 \x01(\x0b\x32\x31.flyteidl.plugins.sagemaker.IntegerParameterRangeH\x00\x12\\\n\x1b\x63\x61tegorical_parameter_range\x18\x03 \x01(\x0b\x32\x35.flyteidl.plugins.sagemaker.CategoricalParameterRangeH\x00\x42\x16\n\x14parameter_range_type\"\xdd\x01\n\x0fParameterRanges\x12_\n\x13parameter_range_map\x18\x01 \x03(\x0b\x32\x42.flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry\x1ai\n\x16ParameterRangeMapEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12>\n\x05value\x18\x02 \x01(\x0b\x32/.flyteidl.plugins.sagemaker.ParameterRangeOneOf:\x02\x38\x01\x42\x35Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/pluginsb\x06proto3') +) + + + +_HYPERPARAMETERSCALINGTYPE_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.HyperparameterScalingType.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='AUTO', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LINEAR', index=1, number=1, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='LOGARITHMIC', index=2, number=2, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='REVERSELOGARITHMIC', index=3, number=3, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=110, + serialized_end=180, +) +_sym_db.RegisterEnumDescriptor(_HYPERPARAMETERSCALINGTYPE_VALUE) + + +_HYPERPARAMETERSCALINGTYPE = _descriptor.Descriptor( + name='HyperparameterScalingType', + full_name='flyteidl.plugins.sagemaker.HyperparameterScalingType', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _HYPERPARAMETERSCALINGTYPE_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=81, + serialized_end=180, +) + + +_CONTINUOUSPARAMETERRANGE = _descriptor.Descriptor( + name='ContinuousParameterRange', + full_name='flyteidl.plugins.sagemaker.ContinuousParameterRange', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='max_value', full_name='flyteidl.plugins.sagemaker.ContinuousParameterRange.max_value', index=0, + number=1, type=1, cpp_type=5, label=1, + has_default_value=False, default_value=float(0), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_value', full_name='flyteidl.plugins.sagemaker.ContinuousParameterRange.min_value', index=1, + number=2, type=1, cpp_type=5, label=1, + has_default_value=False, default_value=float(0), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scaling_type', full_name='flyteidl.plugins.sagemaker.ContinuousParameterRange.scaling_type', index=2, + number=3, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=183, + serialized_end=330, +) + + +_INTEGERPARAMETERRANGE = _descriptor.Descriptor( + name='IntegerParameterRange', + full_name='flyteidl.plugins.sagemaker.IntegerParameterRange', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='max_value', full_name='flyteidl.plugins.sagemaker.IntegerParameterRange.max_value', index=0, + number=1, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='min_value', full_name='flyteidl.plugins.sagemaker.IntegerParameterRange.min_value', index=1, + number=2, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='scaling_type', full_name='flyteidl.plugins.sagemaker.IntegerParameterRange.scaling_type', index=2, + number=3, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=333, + serialized_end=477, +) + + +_CATEGORICALPARAMETERRANGE = _descriptor.Descriptor( + name='CategoricalParameterRange', + full_name='flyteidl.plugins.sagemaker.CategoricalParameterRange', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='values', full_name='flyteidl.plugins.sagemaker.CategoricalParameterRange.values', index=0, + number=1, type=9, cpp_type=9, label=3, + has_default_value=False, default_value=[], + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=479, + serialized_end=522, +) + + +_PARAMETERRANGEONEOF = _descriptor.Descriptor( + name='ParameterRangeOneOf', + full_name='flyteidl.plugins.sagemaker.ParameterRangeOneOf', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='continuous_parameter_range', full_name='flyteidl.plugins.sagemaker.ParameterRangeOneOf.continuous_parameter_range', index=0, + number=1, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='integer_parameter_range', full_name='flyteidl.plugins.sagemaker.ParameterRangeOneOf.integer_parameter_range', index=1, + number=2, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='categorical_parameter_range', full_name='flyteidl.plugins.sagemaker.ParameterRangeOneOf.categorical_parameter_range', index=2, + number=3, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + _descriptor.OneofDescriptor( + name='parameter_range_type', full_name='flyteidl.plugins.sagemaker.ParameterRangeOneOf.parameter_range_type', + index=0, containing_type=None, fields=[]), + ], + serialized_start=525, + serialized_end=842, +) + + +_PARAMETERRANGES_PARAMETERRANGEMAPENTRY = _descriptor.Descriptor( + name='ParameterRangeMapEntry', + full_name='flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='key', full_name='flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.key', index=0, + number=1, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='value', full_name='flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry.value', index=1, + number=2, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=_b('8\001'), + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=961, + serialized_end=1066, +) + +_PARAMETERRANGES = _descriptor.Descriptor( + name='ParameterRanges', + full_name='flyteidl.plugins.sagemaker.ParameterRanges', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='parameter_range_map', full_name='flyteidl.plugins.sagemaker.ParameterRanges.parameter_range_map', index=0, + number=1, type=11, cpp_type=10, label=3, + has_default_value=False, default_value=[], + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[_PARAMETERRANGES_PARAMETERRANGEMAPENTRY, ], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=845, + serialized_end=1066, +) + +_HYPERPARAMETERSCALINGTYPE_VALUE.containing_type = _HYPERPARAMETERSCALINGTYPE +_CONTINUOUSPARAMETERRANGE.fields_by_name['scaling_type'].enum_type = _HYPERPARAMETERSCALINGTYPE_VALUE +_INTEGERPARAMETERRANGE.fields_by_name['scaling_type'].enum_type = _HYPERPARAMETERSCALINGTYPE_VALUE +_PARAMETERRANGEONEOF.fields_by_name['continuous_parameter_range'].message_type = _CONTINUOUSPARAMETERRANGE +_PARAMETERRANGEONEOF.fields_by_name['integer_parameter_range'].message_type = _INTEGERPARAMETERRANGE +_PARAMETERRANGEONEOF.fields_by_name['categorical_parameter_range'].message_type = _CATEGORICALPARAMETERRANGE +_PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'].fields.append( + _PARAMETERRANGEONEOF.fields_by_name['continuous_parameter_range']) +_PARAMETERRANGEONEOF.fields_by_name['continuous_parameter_range'].containing_oneof = _PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'] +_PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'].fields.append( + _PARAMETERRANGEONEOF.fields_by_name['integer_parameter_range']) +_PARAMETERRANGEONEOF.fields_by_name['integer_parameter_range'].containing_oneof = _PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'] +_PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'].fields.append( + _PARAMETERRANGEONEOF.fields_by_name['categorical_parameter_range']) +_PARAMETERRANGEONEOF.fields_by_name['categorical_parameter_range'].containing_oneof = _PARAMETERRANGEONEOF.oneofs_by_name['parameter_range_type'] +_PARAMETERRANGES_PARAMETERRANGEMAPENTRY.fields_by_name['value'].message_type = _PARAMETERRANGEONEOF +_PARAMETERRANGES_PARAMETERRANGEMAPENTRY.containing_type = _PARAMETERRANGES +_PARAMETERRANGES.fields_by_name['parameter_range_map'].message_type = _PARAMETERRANGES_PARAMETERRANGEMAPENTRY +DESCRIPTOR.message_types_by_name['HyperparameterScalingType'] = _HYPERPARAMETERSCALINGTYPE +DESCRIPTOR.message_types_by_name['ContinuousParameterRange'] = _CONTINUOUSPARAMETERRANGE +DESCRIPTOR.message_types_by_name['IntegerParameterRange'] = _INTEGERPARAMETERRANGE +DESCRIPTOR.message_types_by_name['CategoricalParameterRange'] = _CATEGORICALPARAMETERRANGE +DESCRIPTOR.message_types_by_name['ParameterRangeOneOf'] = _PARAMETERRANGEONEOF +DESCRIPTOR.message_types_by_name['ParameterRanges'] = _PARAMETERRANGES +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +HyperparameterScalingType = _reflection.GeneratedProtocolMessageType('HyperparameterScalingType', (_message.Message,), dict( + DESCRIPTOR = _HYPERPARAMETERSCALINGTYPE, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.HyperparameterScalingType) + )) +_sym_db.RegisterMessage(HyperparameterScalingType) + +ContinuousParameterRange = _reflection.GeneratedProtocolMessageType('ContinuousParameterRange', (_message.Message,), dict( + DESCRIPTOR = _CONTINUOUSPARAMETERRANGE, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ContinuousParameterRange) + )) +_sym_db.RegisterMessage(ContinuousParameterRange) + +IntegerParameterRange = _reflection.GeneratedProtocolMessageType('IntegerParameterRange', (_message.Message,), dict( + DESCRIPTOR = _INTEGERPARAMETERRANGE, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.IntegerParameterRange) + )) +_sym_db.RegisterMessage(IntegerParameterRange) + +CategoricalParameterRange = _reflection.GeneratedProtocolMessageType('CategoricalParameterRange', (_message.Message,), dict( + DESCRIPTOR = _CATEGORICALPARAMETERRANGE, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.CategoricalParameterRange) + )) +_sym_db.RegisterMessage(CategoricalParameterRange) + +ParameterRangeOneOf = _reflection.GeneratedProtocolMessageType('ParameterRangeOneOf', (_message.Message,), dict( + DESCRIPTOR = _PARAMETERRANGEONEOF, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRangeOneOf) + )) +_sym_db.RegisterMessage(ParameterRangeOneOf) + +ParameterRanges = _reflection.GeneratedProtocolMessageType('ParameterRanges', (_message.Message,), dict( + + ParameterRangeMapEntry = _reflection.GeneratedProtocolMessageType('ParameterRangeMapEntry', (_message.Message,), dict( + DESCRIPTOR = _PARAMETERRANGES_PARAMETERRANGEMAPENTRY, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRanges.ParameterRangeMapEntry) + )) + , + DESCRIPTOR = _PARAMETERRANGES, + __module__ = 'flyteidl.plugins.sagemaker.parameter_ranges_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.ParameterRanges) + )) +_sym_db.RegisterMessage(ParameterRanges) +_sym_db.RegisterMessage(ParameterRanges.ParameterRangeMapEntry) + + +DESCRIPTOR._options = None +_PARAMETERRANGES_PARAMETERRANGEMAPENTRY._options = None +# @@protoc_insertion_point(module_scope) diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2_grpc.py b/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2_grpc.py new file mode 100644 index 000000000..a89435267 --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/parameter_ranges_pb2_grpc.py @@ -0,0 +1,3 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +import grpc + diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2.py b/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2.py new file mode 100644 index 000000000..776e925b9 --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2.py @@ -0,0 +1,416 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: flyteidl/plugins/sagemaker/training_job.proto + +import sys +_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1')) +from google.protobuf import descriptor as _descriptor +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2 + + +DESCRIPTOR = _descriptor.FileDescriptor( + name='flyteidl/plugins/sagemaker/training_job.proto', + package='flyteidl.plugins.sagemaker', + syntax='proto3', + serialized_options=_b('Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins'), + serialized_pb=_b('\n-flyteidl/plugins/sagemaker/training_job.proto\x12\x1a\x66lyteidl.plugins.sagemaker\x1a\x1egoogle/protobuf/duration.proto\"(\n\tInputMode\"\x1b\n\x05Value\x12\x08\n\x04\x46ILE\x10\x00\x12\x08\n\x04PIPE\x10\x01\"1\n\rAlgorithmName\" \n\x05Value\x12\n\n\x06\x43USTOM\x10\x00\x12\x0b\n\x07XGBOOST\x10\x01\")\n\x10InputContentType\"\x15\n\x05Value\x12\x0c\n\x08TEXT_CSV\x10\x00\"/\n\x10MetricDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05regex\x18\x02 \x01(\t\"\xd7\x02\n\x16\x41lgorithmSpecification\x12?\n\ninput_mode\x18\x01 \x01(\x0e\x32+.flyteidl.plugins.sagemaker.InputMode.Value\x12G\n\x0e\x61lgorithm_name\x18\x02 \x01(\x0e\x32/.flyteidl.plugins.sagemaker.AlgorithmName.Value\x12\x19\n\x11\x61lgorithm_version\x18\x03 \x01(\t\x12H\n\x12metric_definitions\x18\x04 \x03(\x0b\x32,.flyteidl.plugins.sagemaker.MetricDefinition\x12N\n\x12input_content_type\x18\x05 \x01(\x0e\x32\x32.flyteidl.plugins.sagemaker.InputContentType.Value\"e\n\x19TrainingJobResourceConfig\x12\x16\n\x0einstance_count\x18\x01 \x01(\x03\x12\x15\n\rinstance_type\x18\x02 \x01(\t\x12\x19\n\x11volume_size_in_gb\x18\x03 \x01(\x03\"\xbf\x01\n\x0bTrainingJob\x12S\n\x17\x61lgorithm_specification\x18\x01 \x01(\x0b\x32\x32.flyteidl.plugins.sagemaker.AlgorithmSpecification\x12[\n\x1ctraining_job_resource_config\x18\x02 \x01(\x0b\x32\x35.flyteidl.plugins.sagemaker.TrainingJobResourceConfigB5Z3github.com/lyft/flyteidl/gen/pb-go/flyteidl/pluginsb\x06proto3') + , + dependencies=[google_dot_protobuf_dot_duration__pb2.DESCRIPTOR,]) + + + +_INPUTMODE_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.InputMode.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='FILE', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='PIPE', index=1, number=1, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=122, + serialized_end=149, +) +_sym_db.RegisterEnumDescriptor(_INPUTMODE_VALUE) + +_ALGORITHMNAME_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.AlgorithmName.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='CUSTOM', index=0, number=0, + serialized_options=None, + type=None), + _descriptor.EnumValueDescriptor( + name='XGBOOST', index=1, number=1, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=168, + serialized_end=200, +) +_sym_db.RegisterEnumDescriptor(_ALGORITHMNAME_VALUE) + +_INPUTCONTENTTYPE_VALUE = _descriptor.EnumDescriptor( + name='Value', + full_name='flyteidl.plugins.sagemaker.InputContentType.Value', + filename=None, + file=DESCRIPTOR, + values=[ + _descriptor.EnumValueDescriptor( + name='TEXT_CSV', index=0, number=0, + serialized_options=None, + type=None), + ], + containing_type=None, + serialized_options=None, + serialized_start=222, + serialized_end=243, +) +_sym_db.RegisterEnumDescriptor(_INPUTCONTENTTYPE_VALUE) + + +_INPUTMODE = _descriptor.Descriptor( + name='InputMode', + full_name='flyteidl.plugins.sagemaker.InputMode', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _INPUTMODE_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=109, + serialized_end=149, +) + + +_ALGORITHMNAME = _descriptor.Descriptor( + name='AlgorithmName', + full_name='flyteidl.plugins.sagemaker.AlgorithmName', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _ALGORITHMNAME_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=151, + serialized_end=200, +) + + +_INPUTCONTENTTYPE = _descriptor.Descriptor( + name='InputContentType', + full_name='flyteidl.plugins.sagemaker.InputContentType', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + _INPUTCONTENTTYPE_VALUE, + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=202, + serialized_end=243, +) + + +_METRICDEFINITION = _descriptor.Descriptor( + name='MetricDefinition', + full_name='flyteidl.plugins.sagemaker.MetricDefinition', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='name', full_name='flyteidl.plugins.sagemaker.MetricDefinition.name', index=0, + number=1, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='regex', full_name='flyteidl.plugins.sagemaker.MetricDefinition.regex', index=1, + number=2, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=245, + serialized_end=292, +) + + +_ALGORITHMSPECIFICATION = _descriptor.Descriptor( + name='AlgorithmSpecification', + full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='input_mode', full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification.input_mode', index=0, + number=1, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='algorithm_name', full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_name', index=1, + number=2, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='algorithm_version', full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification.algorithm_version', index=2, + number=3, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='metric_definitions', full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification.metric_definitions', index=3, + number=4, type=11, cpp_type=10, label=3, + has_default_value=False, default_value=[], + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='input_content_type', full_name='flyteidl.plugins.sagemaker.AlgorithmSpecification.input_content_type', index=4, + number=5, type=14, cpp_type=8, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=295, + serialized_end=638, +) + + +_TRAININGJOBRESOURCECONFIG = _descriptor.Descriptor( + name='TrainingJobResourceConfig', + full_name='flyteidl.plugins.sagemaker.TrainingJobResourceConfig', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='instance_count', full_name='flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_count', index=0, + number=1, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='instance_type', full_name='flyteidl.plugins.sagemaker.TrainingJobResourceConfig.instance_type', index=1, + number=2, type=9, cpp_type=9, label=1, + has_default_value=False, default_value=_b("").decode('utf-8'), + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='volume_size_in_gb', full_name='flyteidl.plugins.sagemaker.TrainingJobResourceConfig.volume_size_in_gb', index=2, + number=3, type=3, cpp_type=2, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=640, + serialized_end=741, +) + + +_TRAININGJOB = _descriptor.Descriptor( + name='TrainingJob', + full_name='flyteidl.plugins.sagemaker.TrainingJob', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='algorithm_specification', full_name='flyteidl.plugins.sagemaker.TrainingJob.algorithm_specification', index=0, + number=1, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + _descriptor.FieldDescriptor( + name='training_job_resource_config', full_name='flyteidl.plugins.sagemaker.TrainingJob.training_job_resource_config', index=1, + number=2, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto3', + extension_ranges=[], + oneofs=[ + ], + serialized_start=744, + serialized_end=935, +) + +_INPUTMODE_VALUE.containing_type = _INPUTMODE +_ALGORITHMNAME_VALUE.containing_type = _ALGORITHMNAME +_INPUTCONTENTTYPE_VALUE.containing_type = _INPUTCONTENTTYPE +_ALGORITHMSPECIFICATION.fields_by_name['input_mode'].enum_type = _INPUTMODE_VALUE +_ALGORITHMSPECIFICATION.fields_by_name['algorithm_name'].enum_type = _ALGORITHMNAME_VALUE +_ALGORITHMSPECIFICATION.fields_by_name['metric_definitions'].message_type = _METRICDEFINITION +_ALGORITHMSPECIFICATION.fields_by_name['input_content_type'].enum_type = _INPUTCONTENTTYPE_VALUE +_TRAININGJOB.fields_by_name['algorithm_specification'].message_type = _ALGORITHMSPECIFICATION +_TRAININGJOB.fields_by_name['training_job_resource_config'].message_type = _TRAININGJOBRESOURCECONFIG +DESCRIPTOR.message_types_by_name['InputMode'] = _INPUTMODE +DESCRIPTOR.message_types_by_name['AlgorithmName'] = _ALGORITHMNAME +DESCRIPTOR.message_types_by_name['InputContentType'] = _INPUTCONTENTTYPE +DESCRIPTOR.message_types_by_name['MetricDefinition'] = _METRICDEFINITION +DESCRIPTOR.message_types_by_name['AlgorithmSpecification'] = _ALGORITHMSPECIFICATION +DESCRIPTOR.message_types_by_name['TrainingJobResourceConfig'] = _TRAININGJOBRESOURCECONFIG +DESCRIPTOR.message_types_by_name['TrainingJob'] = _TRAININGJOB +_sym_db.RegisterFileDescriptor(DESCRIPTOR) + +InputMode = _reflection.GeneratedProtocolMessageType('InputMode', (_message.Message,), dict( + DESCRIPTOR = _INPUTMODE, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputMode) + )) +_sym_db.RegisterMessage(InputMode) + +AlgorithmName = _reflection.GeneratedProtocolMessageType('AlgorithmName', (_message.Message,), dict( + DESCRIPTOR = _ALGORITHMNAME, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmName) + )) +_sym_db.RegisterMessage(AlgorithmName) + +InputContentType = _reflection.GeneratedProtocolMessageType('InputContentType', (_message.Message,), dict( + DESCRIPTOR = _INPUTCONTENTTYPE, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.InputContentType) + )) +_sym_db.RegisterMessage(InputContentType) + +MetricDefinition = _reflection.GeneratedProtocolMessageType('MetricDefinition', (_message.Message,), dict( + DESCRIPTOR = _METRICDEFINITION, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.MetricDefinition) + )) +_sym_db.RegisterMessage(MetricDefinition) + +AlgorithmSpecification = _reflection.GeneratedProtocolMessageType('AlgorithmSpecification', (_message.Message,), dict( + DESCRIPTOR = _ALGORITHMSPECIFICATION, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.AlgorithmSpecification) + )) +_sym_db.RegisterMessage(AlgorithmSpecification) + +TrainingJobResourceConfig = _reflection.GeneratedProtocolMessageType('TrainingJobResourceConfig', (_message.Message,), dict( + DESCRIPTOR = _TRAININGJOBRESOURCECONFIG, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJobResourceConfig) + )) +_sym_db.RegisterMessage(TrainingJobResourceConfig) + +TrainingJob = _reflection.GeneratedProtocolMessageType('TrainingJob', (_message.Message,), dict( + DESCRIPTOR = _TRAININGJOB, + __module__ = 'flyteidl.plugins.sagemaker.training_job_pb2' + # @@protoc_insertion_point(class_scope:flyteidl.plugins.sagemaker.TrainingJob) + )) +_sym_db.RegisterMessage(TrainingJob) + + +DESCRIPTOR._options = None +# @@protoc_insertion_point(module_scope) diff --git a/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2_grpc.py b/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2_grpc.py new file mode 100644 index 000000000..a89435267 --- /dev/null +++ b/gen/pb_python/flyteidl/plugins/sagemaker/training_job_pb2_grpc.py @@ -0,0 +1,3 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +import grpc + diff --git a/generate_protos.sh b/generate_protos.sh index 889e93b8f..6822fd20a 100755 --- a/generate_protos.sh +++ b/generate_protos.sh @@ -2,7 +2,7 @@ DIR=`pwd` rm -rf $DIR/gen -LYFT_IMAGE="lyft/protocgenerator:5e6a3be18db77a8862365a19711428c2f66284ef" +LYFT_IMAGE="lyft/protocgenerator:8167e11d3b3439373c2f033080a4b550078884a2" SWAGGER_CLI_IMAGE="docker.io/lyft/swagger-codegen-cli:dc5ce6ec6d7d4d980fa882d6bd13a83cba3be3c3" docker run --rm -u $(id -u):$(id -g) -v $DIR:/defs $LYFT_IMAGE -i ./protos -d protos/flyteidl/service --with_gateway -l go --go_source_relative diff --git a/package.json b/package.json index f1df11889..4b36f9959 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "@lyft/flyteidl", - "version": "0.17.38", + "version": "0.18.0", "description": "Compiled protocol buffers and gRPC service clients/servers for Flyte IDLs", "repository": { "type": "git", diff --git a/protos/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto b/protos/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto new file mode 100644 index 000000000..bfc3b1400 --- /dev/null +++ b/protos/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto @@ -0,0 +1,83 @@ +syntax = "proto3"; + +package flyteidl.plugins.sagemaker; + +option go_package = "github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins"; + +import "flyteidl/plugins/sagemaker/parameter_ranges.proto"; +import "flyteidl/plugins/sagemaker/training_job.proto"; + +// A pass-through for SageMaker's hyperparameter tuning job +message HyperparameterTuningJob { + // The underlying training job that the hyperparameter tuning job will launch during the process + TrainingJob training_job = 1; + + // The maximum number of training jobs that an hpo job can launch. For resource limit purpose. + // https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + int64 max_number_of_training_jobs = 2; + + // The maximum number of concurrent training job that an hpo job can launch + // https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ResourceLimits.html + int64 max_parallel_training_jobs = 3; +} + +// HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job +// with respect to the specified metric. +message HyperparameterTuningObjectiveType { + enum Value { + MINIMIZE = 0; + MAXIMIZE = 1; + } +} + +// The target metric and the objective of the hyperparameter tuning. +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html +message HyperparameterTuningObjective { + + // HyperparameterTuningObjectiveType determines the direction of the tuning of the Hyperparameter Tuning Job + // with respect to the specified metric. + HyperparameterTuningObjectiveType.Value objective_type = 1; + + // The target metric name, which is the user-defined name of the metric specified in the + // training job's algorithm specification + string metric_name = 2; +} + + +// Setting the strategy used when searching in the hyperparameter space +// Refer this doc for more details: +// https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ +message HyperparameterTuningStrategy { + enum Value { + BAYESIAN = 0; + RANDOM = 1; + } +} + +// When the training jobs launched by the hyperparameter tuning job are not improving significantly, +// a hyperparameter tuning job can be stopping early. +// Note that there's only a subset of built-in algorithms that supports early stopping. +// see: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-early-stopping.html +message TrainingJobEarlyStoppingType { + enum Value { + OFF = 0; + AUTO = 1; + } +} + +// The specification of the hyperparameter tuning process +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-ex-tuning-job.html#automatic-model-tuning-ex-low-tuning-config +message HyperparameterTuningJobConfig { + // ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range + ParameterRanges hyperparameter_ranges = 1; + + // Setting the strategy used when searching in the hyperparameter space + HyperparameterTuningStrategy.Value tuning_strategy = 2; + + // The target metric and the objective of the hyperparameter tuning. + HyperparameterTuningObjective tuning_objective = 3; + + // When the training jobs launched by the hyperparameter tuning job are not improving significantly, + // a hyperparameter tuning job can be stopping early. + TrainingJobEarlyStoppingType.Value training_job_early_stopping_type = 4; +} diff --git a/protos/flyteidl/plugins/sagemaker/parameter_ranges.proto b/protos/flyteidl/plugins/sagemaker/parameter_ranges.proto new file mode 100644 index 000000000..909f4e6eb --- /dev/null +++ b/protos/flyteidl/plugins/sagemaker/parameter_ranges.proto @@ -0,0 +1,61 @@ +syntax = "proto3"; + +package flyteidl.plugins.sagemaker; + +option go_package = "github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins"; + +// HyperparameterScalingType defines the way to increase or decrease the value of the hyperparameter +// For details, refer to: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +// See examples of these scaling type, refer to: https://aws.amazon.com/blogs/machine-learning/amazon-sagemaker-automatic-model-tuning-now-supports-random-search-and-hyperparameter-scaling/ +message HyperparameterScalingType { + enum Value { + AUTO = 0; + LINEAR = 1; + LOGARITHMIC = 2; + REVERSELOGARITHMIC = 3; + } +} + +// ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +// users to specify the search space of a floating-point hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +message ContinuousParameterRange { + double max_value = 1; + double min_value = 2; + HyperparameterScalingType.Value scaling_type = 3; +} + +// IntegerParameterRange refers to a discrete range of hyperparameter values, allowing +// users to specify the search space of an integer hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +message IntegerParameterRange { + int64 max_value = 1; + int64 min_value = 2; + HyperparameterScalingType.Value scaling_type = 3; +} + +// ContinuousParameterRange refers to a continuous range of hyperparameter values, allowing +// users to specify the search space of a floating-point hyperparameter +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +message CategoricalParameterRange { + repeated string values = 1; +} + + +// ParameterRangeOneOf describes a single ParameterRange, which is a one-of structure that can be one of +// the three possible types: ContinuousParameterRange, IntegerParameterRange, and CategoricalParameterRange. +// This one-of structure in Flyte enables specifying a Parameter in a type-safe manner +// See: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +message ParameterRangeOneOf { + oneof parameter_range_type { + ContinuousParameterRange continuous_parameter_range = 1; + IntegerParameterRange integer_parameter_range = 2; + CategoricalParameterRange categorical_parameter_range = 3; + } +} + +// ParameterRanges is a map that maps hyperparameter name to the corresponding hyperparameter range +// https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-ranges.html +message ParameterRanges { + map parameter_range_map = 1; +} diff --git a/protos/flyteidl/plugins/sagemaker/training_job.proto b/protos/flyteidl/plugins/sagemaker/training_job.proto new file mode 100644 index 000000000..1d8dc820d --- /dev/null +++ b/protos/flyteidl/plugins/sagemaker/training_job.proto @@ -0,0 +1,100 @@ +syntax = "proto3"; + +package flyteidl.plugins.sagemaker; + +import "google/protobuf/duration.proto"; + +option go_package = "github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins"; + +// The input mode that the algorithm supports. When using the File input mode, SageMaker downloads +// the training data from S3 to the provisioned ML storage Volume, and mounts the directory to docker +// volume for training container. When using Pipe input mode, Amazon SageMaker streams data directly +// from S3 to the container. +// See: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html +// For the input modes that different SageMaker algorithms support, see: +// https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html +message InputMode { + enum Value { + FILE = 0; + PIPE = 1; + } +} + +// The algorithm name is used for deciding which pre-built image to point to. +// This is only required for use cases where SageMaker's built-in algorithm mode is used. +// While we currently only support a subset of the algorithms, more will be added to the list. +// See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html +message AlgorithmName { + enum Value { + CUSTOM = 0; + XGBOOST = 1; + } +} + + +// Specifies the type of file for input data. Different SageMaker built-in algorithms require different file types of input data +// See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html +// https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html +message InputContentType { + enum Value { + TEXT_CSV = 0; + } +} + +// Specifies a metric that the training algorithm writes to stderr or stdout. +// This object is a pass-through. +// See this for details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_MetricDefinition.html +message MetricDefinition { + // User-defined name of the metric + string name = 1; + // SageMaker hyperparameter tuning parses your algorithm’s stdout and stderr streams to find algorithm metrics + string regex = 2; +} + + +// Specifies the training algorithm to be used in the training job +// This object is mostly a pass-through, with a couple of exceptions include: (1) in Flyte, users don't need to specify +// TrainingImage; either use the built-in algorithm mode by using Flytekit's Simple Training Job and specifying an algorithm +// name and an algorithm version or (2) when users want to supply custom algorithms they should set algorithm_name field to +// CUSTOM. In this case, the value of the algorithm_version field has no effect +// For pass-through use cases: refer to this AWS official document for more details +// https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html +message AlgorithmSpecification { + // The input mode can be either PIPE or FILE + InputMode.Value input_mode = 1; + + // The algorithm name is used for deciding which pre-built image to point to + AlgorithmName.Value algorithm_name = 2; + // The algorithm version field is used for deciding which pre-built image to point to + // This is only needed for use cases where SageMaker's built-in algorithm mode is chosen + string algorithm_version = 3; + + // A list of metric definitions for SageMaker to evaluate/track on the progress of the training job + // See this: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_AlgorithmSpecification.html + // and this: https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-define-metrics.html + repeated MetricDefinition metric_definitions = 4; + + // The content type of the input + // See https://docs.aws.amazon.com/sagemaker/latest/dg/cdf-training.html + // https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-algo-docker-registry-paths.html + InputContentType.Value input_content_type = 5; +} + +// TrainingJobResourceConfig is a pass-through, specifying the instance type to use for the training job, the +// number of instances to launch, and the size of the ML storage volume the user wants to provision +// Refer to SageMaker official doc for more details: https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html +message TrainingJobResourceConfig { + // The number of ML compute instances to use. For distributed training, provide a value greater than 1. + int64 instance_count = 1; + // The ML compute instance type + string instance_type = 2; + // The size of the ML storage volume that you want to provision. + int64 volume_size_in_gb = 3; +} + +// The spec of a training job. This is mostly a pass-through object +// https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateTrainingJob.html +message TrainingJob { + AlgorithmSpecification algorithm_specification = 1; + TrainingJobResourceConfig training_job_resource_config = 2; +} diff --git a/setup.py b/setup.py index 78814088f..ee6260fd0 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,6 @@ from setuptools import setup, find_packages -__version__ = '0.17.38' +__version__ = '0.18.0' setup( name='flyteidl',