-
Notifications
You must be signed in to change notification settings - Fork 2.2k
/
notebook_execution_job.proto
138 lines (115 loc) · 5.19 KB
/
notebook_execution_job.proto
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
// Copyright 2023 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
syntax = "proto3";
package google.cloud.aiplatform.v1beta1;
import "google/api/field_behavior.proto";
import "google/api/resource.proto";
import "google/cloud/aiplatform/v1beta1/job_state.proto";
import "google/protobuf/duration.proto";
import "google/protobuf/timestamp.proto";
import "google/rpc/status.proto";
option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
option go_package = "cloud.google.com/go/aiplatform/apiv1beta1/aiplatformpb;aiplatformpb";
option java_multiple_files = true;
option java_outer_classname = "NotebookExecutionJobProto";
option java_package = "com.google.cloud.aiplatform.v1beta1";
option php_namespace = "Google\\Cloud\\AIPlatform\\V1beta1";
option ruby_package = "Google::Cloud::AIPlatform::V1beta1";
// NotebookExecutionJob represents an instance of a notebook execution.
message NotebookExecutionJob {
option (google.api.resource) = {
type: "aiplatform.googleapis.com/NotebookExecutionJob"
pattern: "projects/{project}/locations/{location}/notebookExecutionJobs/{notebook_execution_job}"
plural: "notebookExecutionJobs"
singular: "notebookExecutionJob"
};
// The Dataform Repository containing the input notebook.
message DataformRepositorySource {
// The resource name of the Dataform Repository. Format:
// `projects/{project_id}/locations/{location}/repositories/{repository_id}`
string dataform_repository_resource_name = 1;
// The commit SHA to read repository with. If unset, the file will be read
// at HEAD.
string commit_sha = 2;
}
// The Cloud Storage uri for the input notebook.
message GcsNotebookSource {
// The Cloud Storage uri pointing to the ipynb file. Format:
// `gs://bucket/notebook_file.ipynb`
string uri = 1;
// The version of the Cloud Storage object to read. If unset, the current
// version of the object is read. See
// https://cloud.google.com/storage/docs/metadata#generation-number.
string generation = 2;
}
// The input notebook.
oneof notebook_source {
// The Dataform Repository pointing to a single file notebook repository.
DataformRepositorySource dataform_repository_source = 3;
// The GCS url pointing to the ipynb file. Format:
// `gs://bucket/notebook_file.ipynb`
GcsNotebookSource gcs_notebook_source = 4;
}
// The compute config to use for an execution job.
oneof environment_spec {
// The NotebookRuntimeTemplate to source compute configuration from.
string notebook_runtime_template_resource_name = 14
[(google.api.resource_reference) = {
type: "aiplatform.googleapis.com/NotebookRuntimeTemplate"
}];
}
// The location to store the notebook execution result.
oneof execution_sink {
// The GCS location to upload the result to. Format:
// `gs://bucket-name`
string gcs_output_uri = 8;
}
// The identity to run the execution as.
oneof execution_identity {
// The user email to run the execution as. Only supported by Colab runtimes.
string execution_user = 9;
// The service account to run the execution as.
string service_account = 18;
}
// Output only. The resource name of this NotebookExecutionJob. Format:
// `projects/{project_id}/locations/{location}/notebookExecutionJobs/{job_id}`
string name = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
// The display name of the NotebookExecutionJob. The name can be up to 128
// characters long and can consist of any UTF-8 characters.
string display_name = 2;
// Max running time of the execution job in seconds (default 86400s / 24 hrs).
google.protobuf.Duration execution_timeout = 5;
// Output only. The Schedule resource name if this job is triggered by one.
// Format:
// `projects/{project_id}/locations/{location}/schedules/{schedule_id}`
string schedule_resource_name = 6 [
(google.api.field_behavior) = OUTPUT_ONLY,
(google.api.resource_reference) = {
type: "aiplatform.googleapis.com/Schedule"
}
];
// Output only. The state of the NotebookExecutionJob.
JobState job_state = 10 [(google.api.field_behavior) = OUTPUT_ONLY];
// Output only. Populated when the NotebookExecutionJob is completed. When
// there is an error during notebook execution, the error details are
// populated.
google.rpc.Status status = 11 [(google.api.field_behavior) = OUTPUT_ONLY];
// Output only. Timestamp when this NotebookExecutionJob was created.
google.protobuf.Timestamp create_time = 12
[(google.api.field_behavior) = OUTPUT_ONLY];
// Output only. Timestamp when this NotebookExecutionJob was most recently
// updated.
google.protobuf.Timestamp update_time = 13
[(google.api.field_behavior) = OUTPUT_ONLY];
}