@flyteorg/flyteidl 1.5.21 → 1.10.7

package/protos/flyteidl/service/agent.proto

@@ -1,10 +1,12 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
+
+import "google/api/annotations.proto";
 import "flyteidl/admin/agent.proto";
 
-// AgentService defines an RPC Service that allows propeller to send the request to the agent server.
+// AsyncAgentService defines an RPC Service that allows propeller to send the request to the agent server.
 service AsyncAgentService {
   // Send a task create request to the agent server.
   rpc CreateTask (flyteidl.admin.CreateTaskRequest) returns (flyteidl.admin.CreateTaskResponse){};
@@ -12,4 +14,32 @@ service AsyncAgentService {
   rpc GetTask (flyteidl.admin.GetTaskRequest) returns (flyteidl.admin.GetTaskResponse){};
   // Delete the task resource.
   rpc DeleteTask (flyteidl.admin.DeleteTaskRequest) returns (flyteidl.admin.DeleteTaskResponse){};
+
+  // GetTaskMetrics returns one or more task execution metrics, if available.
+  //
+  // Errors include
+  //  * OutOfRange if metrics are not available for the specified task time range
+  //  * various other errors
+  rpc GetTaskMetrics(flyteidl.admin.GetTaskMetricsRequest) returns (flyteidl.admin.GetTaskMetricsResponse){};
+
+  // GetTaskLogs returns task execution logs, if available.
+  rpc GetTaskLogs(flyteidl.admin.GetTaskLogsRequest) returns (flyteidl.admin.GetTaskLogsResponse){};
+}
+
+// AgentMetadataService defines an RPC service that is also served over HTTP via grpc-gateway.
+// This service allows propeller or users to get the metadata of agents.
+service AgentMetadataService {
+  // Fetch a :ref:`ref_flyteidl.admin.Agent` definition.
+  rpc GetAgent (flyteidl.admin.GetAgentRequest) returns (flyteidl.admin.GetAgentResponse){
+    option (google.api.http) = {
+      get: "/api/v1/agent/{name}" 
+    };
+  };
+
+  // Fetch a list of :ref:`ref_flyteidl.admin.Agent` definitions.
+  rpc ListAgents (flyteidl.admin.ListAgentsRequest) returns (flyteidl.admin.ListAgentsResponse){
+    option (google.api.http) = {
+      get: "/api/v1/agents"
+    };
+  };
 }

package/protos/flyteidl/service/auth.proto

@@ -1,10 +1,10 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
 
 import "google/api/annotations.proto";
-// import "protoc-gen-swagger/options/annotations.proto";
+import "protoc-gen-openapiv2/options/annotations.proto";
 
 message OAuth2MetadataRequest {}
 
@@ -76,9 +76,9 @@ service AuthMetadataService {
         option (google.api.http) = {
             get: "/.well-known/oauth-authorization-server"
         };
-        // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-        //     description: "Retrieves OAuth2 authorization server metadata. This endpoint is anonymously accessible."
-        // };
+        option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+            description: "Retrieves OAuth2 authorization server metadata. This endpoint is anonymously accessible."
+        };
     }
 
     // Anonymously accessible. Retrieves the client information clients should use when initiating OAuth2 authorization
@@ -87,8 +87,8 @@ service AuthMetadataService {
         option (google.api.http) = {
             get: "/config/v1/flyte_client"
         };
-        // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-        //     description: "Retrieves public flyte client info. This endpoint is anonymously accessible."
-        // };
+        option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+            description: "Retrieves public flyte client info. This endpoint is anonymously accessible."
+        };
     }
 }

package/protos/flyteidl/service/dataproxy.proto

@@ -1,10 +1,10 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
 
 import "google/api/annotations.proto";
-// import "protoc-gen-swagger/options/annotations.proto";
+import "protoc-gen-openapiv2/options/annotations.proto";
 import "google/protobuf/duration.proto";
 import "google/protobuf/timestamp.proto";
 import "flyteidl/core/identifier.proto";
@@ -129,7 +129,7 @@ message PreSignedURLs {
 // General request artifact to retrieve data from a Flyte artifact url.
 message GetDataRequest {
   // A unique identifier in the form of flyte://<something> that uniquely, for a given Flyte
-  // backend, identifies a Flyte artifact ([i]nput, [o]utput, flyte [d]eck, etc.).
+  // backend, identifies a Flyte artifact ([i]nput, [o]output, flyte [d]eck, etc.).
   // e.g. flyte://v1/proj/development/execid/n2/0/i (for 0th task execution attempt input)
   //      flyte://v1/proj/development/execid/n2/i (for node execution input)
   //      flyte://v1/proj/development/execid/n2/o/o3 (the o3 output of the second node)
@@ -158,9 +158,9 @@ service DataProxyService {
       post: "/api/v1/dataproxy/artifact_urn"
       body: "*"
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Creates a write-only http location that is accessible for tasks at runtime."
-    // };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Creates a write-only http location that is accessible for tasks at runtime."
+    };
   }
 
   // CreateDownloadLocation creates a signed url to download artifacts.
@@ -169,9 +169,9 @@ service DataProxyService {
     option (google.api.http) = {
       get: "/api/v1/dataproxy/artifact_urn"
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Deprecated: Please use CreateDownloadLink instead. Creates a read-only http location that is accessible for tasks at runtime."
-    // };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Deprecated: Please use CreateDownloadLink instead. Creates a read-only http location that is accessible for tasks at runtime."
+    };
   }
 
   // CreateDownloadLocation creates a signed url to download artifacts.
@@ -180,9 +180,9 @@ service DataProxyService {
       post: "/api/v1/dataproxy/artifact_link"
       body: "*"
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Creates a read-only http location that is accessible for tasks at runtime."
-    // };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Creates a read-only http location that is accessible for tasks at runtime."
+    };
   }
 
   rpc GetData (GetDataRequest) returns (GetDataResponse) {

package/protos/flyteidl/service/external_plugin_service.proto

@@ -1,10 +1,9 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
 import "flyteidl/core/literals.proto";
 import "flyteidl/core/tasks.proto";
-import "flyteidl/core/interface.proto";
 
 // ExternalPluginService defines an RPC Service that allows propeller to send the request to the backend plugin server.
 service ExternalPluginService {

package/protos/flyteidl/service/identity.proto

@@ -1,11 +1,11 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
 
 import "google/api/annotations.proto";
 import "google/protobuf/struct.proto";
-// import "protoc-gen-swagger/options/annotations.proto";
+import "protoc-gen-openapiv2/options/annotations.proto";
 
 message UserInfoRequest {}
 
@@ -44,8 +44,8 @@ service IdentityService {
     option (google.api.http) = {
       get: "/me"
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Retrieves authenticated identity info."
-    // };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Retrieves authenticated identity info."
+    };
   }
 }

package/protos/flyteidl/service/signal.proto

@@ -1,11 +1,11 @@
 syntax = "proto3";
 package flyteidl.service;
 
-option go_package = "github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/service";
+option go_package = "github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/service";
 
 import "google/api/annotations.proto";
 import "flyteidl/admin/signal.proto";
-// import "protoc-gen-swagger/options/annotations.proto";
+import "protoc-gen-openapiv2/options/annotations.proto";
 
 // SignalService defines an RPC Service that may create, update, and retrieve signal(s).
 service SignalService {
@@ -15,19 +15,22 @@ service SignalService {
     // a signal, meaning the first call will create the signal and all subsequent calls will
     // fetch the existing signal. This is only useful during Flyte Workflow execution and therefore
     // is not exposed to mitigate unintended behavior.
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Retrieve a signal, creating it if it does not exist."
-    // };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Retrieve a signal, creating it if it does not exist."
+    };
   }
 
   // Fetch a list of :ref:`ref_flyteidl.admin.Signal` definitions.
   rpc ListSignals (flyteidl.admin.SignalListRequest) returns (flyteidl.admin.SignalList) {
     option (google.api.http) = {
       get: "/api/v1/signals/{workflow_execution_id.project}/{workflow_execution_id.domain}/{workflow_execution_id.name}"
+      additional_bindings: {
+        get: "/api/v1/signals/org/{workflow_execution_id.org}/{workflow_execution_id.project}/{workflow_execution_id.domain}/{workflow_execution_id.name}"
+      }
+    };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Fetch existing signal definitions matching the input signal id filters."
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Fetch existing signal definitions matching the input signal id filters."
-    // };
   }
 
   // Sets the value on a :ref:`ref_flyteidl.admin.Signal` definition
@@ -35,21 +38,25 @@ service SignalService {
     option (google.api.http) = {
       post: "/api/v1/signals"
       body: "*"
+      additional_bindings: {
+        post: "/api/v1/signals/org/{id.execution_id.org}"
+        body: "*"
+      }
+    };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Set a signal value."
+      responses: {
+        key: "400"
+        value: {
+          description: "Returned for bad request that may have failed validation."
+        }
+      }
+      responses: {
+        key: "409"
+        value: {
+          description: "Returned for a request that references an identical entity that has already been registered."
+        }
+      }
     };
-    // option (grpc.gateway.protoc_gen_swagger.options.openapiv2_operation) = {
-    //   description: "Set a signal value."
-    //   responses: {
-    //     key: "400"
-    //     value: {
-    //       description: "Returned for bad request that may have failed validation."
-    //     }
-    //   }
-    //   responses: {
-    //     key: "409"
-    //     value: {
-    //       description: "Returned for a request that references an identical entity that has already been registered."
-    //     }
-    //   }
-    // };
   }
 }

package/protos/flyteidl/plugins/sagemaker/hyperparameter_tuning_job.proto

@@ -1,83 +0,0 @@
-syntax = "proto3";
-
-package flyteidl.plugins.sagemaker;
-
-option go_package = "github.com/flyteorg/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;
-}

package/protos/flyteidl/plugins/sagemaker/parameter_ranges.proto

@@ -1,61 +0,0 @@
-syntax = "proto3";
-
-package flyteidl.plugins.sagemaker;
-
-option go_package = "github.com/flyteorg/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<string, ParameterRangeOneOf> parameter_range_map = 1;
-}

package/protos/flyteidl/plugins/sagemaker/training_job.proto

@@ -1,119 +0,0 @@
-syntax = "proto3";
-
-package flyteidl.plugins.sagemaker;
-
-import "google/protobuf/duration.proto";
-
-option go_package = "github.com/flyteorg/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;
-}
-
-// When enabling distributed training on a training job, the user should use this message to tell Flyte and SageMaker
-// what kind of distributed protocol he/she wants to use to distribute the work.
-message DistributedProtocol {
-    enum Value {
-        // Use this value if the user wishes to use framework-native distributed training interfaces.
-        // If this value is used, Flyte won't configure SageMaker to initialize unnecessary components such as
-        // OpenMPI or Parameter Server.
-        UNSPECIFIED = 0;
-        // Use this value if the user wishes to use MPI as the underlying protocol for her distributed training job
-        // MPI is a framework-agnostic distributed protocol. It has multiple implementations. Currently, we have only
-        // tested the OpenMPI implementation, which is the recommended implementation for Horovod.
-        MPI = 1;
-    }
-}
-
-// 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;
-    // When users specify an instance_count > 1, Flyte will try to configure SageMaker to enable distributed training.
-    // If the users wish to use framework-agnostic distributed protocol such as MPI or Parameter Server, this
-    // field should be set to the corresponding enum value
-    DistributedProtocol.Value distributed_protocol = 4;
-}
-
-// 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;
-}