/
__init__.py
280 lines (269 loc) · 13.6 KB
/
__init__.py
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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
# -*- coding: utf-8 -*-
# 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.
#
from google.cloud.aiplatform.compat import services
from google.cloud.aiplatform.compat import types
V1BETA1 = "v1beta1"
V1 = "v1"
DEFAULT_VERSION = V1
if DEFAULT_VERSION == V1BETA1:
services.dataset_service_client = services.dataset_service_client_v1beta1
services.deployment_resource_pool_service_client = (
services.deployment_resource_pool_service_client_v1beta1
)
services.endpoint_service_client = services.endpoint_service_client_v1beta1
services.feature_online_store_admin_service_client = (
services.feature_online_store_admin_service_client_v1beta1
)
services.feature_online_store_service_client = (
services.feature_online_store_service_client_v1beta1
)
services.feature_registry_service_client = (
services.feature_registry_service_client_v1beta1
)
services.featurestore_online_serving_service_client = (
services.featurestore_online_serving_service_client_v1beta1
)
services.featurestore_service_client = services.featurestore_service_client_v1beta1
services.job_service_client = services.job_service_client_v1beta1
services.model_service_client = services.model_service_client_v1beta1
services.model_garden_service_client = services.model_garden_service_client_v1beta1
services.pipeline_service_client = services.pipeline_service_client_v1beta1
services.prediction_service_client = services.prediction_service_client_v1beta1
services.prediction_service_async_client = (
services.prediction_service_async_client_v1beta1
)
services.schedule_service_client = services.schedule_service_client_v1beta1
services.specialist_pool_service_client = (
services.specialist_pool_service_client_v1beta1
)
services.match_service_client = services.match_service_client_v1beta1
services.metadata_service_client = services.metadata_service_client_v1beta1
services.tensorboard_service_client = services.tensorboard_service_client_v1beta1
services.index_service_client = services.index_service_client_v1beta1
services.index_endpoint_service_client = (
services.index_endpoint_service_client_v1beta1
)
services.vizier_service_client = services.vizier_service_client_v1beta1
types.accelerator_type = types.accelerator_type_v1beta1
types.annotation = types.annotation_v1beta1
types.annotation_spec = types.annotation_spec_v1beta1
types.artifact = types.artifact_v1beta1
types.batch_prediction_job = types.batch_prediction_job_v1beta1
types.completion_stats = types.completion_stats_v1beta1
types.context = types.context_v1beta1
types.custom_job = types.custom_job_v1beta1
types.data_item = types.data_item_v1beta1
types.data_labeling_job = types.data_labeling_job_v1beta1
types.dataset = types.dataset_v1beta1
types.dataset_service = types.dataset_service_v1beta1
types.deployed_model_ref = types.deployed_model_ref_v1beta1
types.deployment_resource_pool = types.deployment_resource_pool_v1beta1
types.encryption_spec = types.encryption_spec_v1beta1
types.endpoint = types.endpoint_v1beta1
types.endpoint_service = types.endpoint_service_v1beta1
types.entity_type = types.entity_type_v1beta1
types.env_var = types.env_var_v1beta1
types.event = types.event_v1beta1
types.execution = types.execution_v1beta1
types.explanation = types.explanation_v1beta1
types.explanation_metadata = types.explanation_metadata_v1beta1
types.feature = types.feature_v1beta1
types.feature_group = types.feature_group_v1beta1
types.feature_monitoring_stats = types.feature_monitoring_stats_v1beta1
types.feature_online_store = types.feature_online_store_v1beta1
types.feature_online_store_admin_service = (
types.feature_online_store_admin_service_v1beta1
)
types.feature_registry_service = types.feature_registry_service_v1beta1
types.feature_online_store_service = types.feature_online_store_service_v1beta1
types.feature_selector = types.feature_selector_v1beta1
types.feature_view = types.feature_view_v1beta1
types.feature_view_sync = types.feature_view_sync_v1beta1
types.featurestore = types.featurestore_v1beta1
types.featurestore_monitoring = types.featurestore_monitoring_v1beta1
types.featurestore_online_service = types.featurestore_online_service_v1beta1
types.featurestore_service = types.featurestore_service_v1beta1
types.hyperparameter_tuning_job = types.hyperparameter_tuning_job_v1beta1
types.index = types.index_v1beta1
types.index_endpoint = types.index_endpoint_v1beta1
types.index_service = types.index_service_v1beta1
types.io = types.io_v1beta1
types.job_service = types.job_service_v1beta1
types.job_state = types.job_state_v1beta1
types.lineage_subgraph = types.lineage_subgraph_v1beta1
types.machine_resources = types.machine_resources_v1beta1
types.manual_batch_tuning_parameters = types.manual_batch_tuning_parameters_v1beta1
types.matching_engine_deployed_index_ref = (
types.matching_engine_deployed_index_ref_v1beta1
)
types.matching_engine_index = types.index_v1beta1
types.matching_engine_index_endpoint = types.index_endpoint_v1beta1
types.metadata_service = types.metadata_service_v1beta1
types.metadata_schema = types.metadata_schema_v1beta1
types.metadata_store = types.metadata_store_v1beta1
types.model = types.model_v1beta1
types.model_evaluation = types.model_evaluation_v1beta1
types.model_evaluation_slice = types.model_evaluation_slice_v1beta1
types.model_deployment_monitoring_job = (
types.model_deployment_monitoring_job_v1beta1
)
types.model_garden_service = types.model_garden_service_v1beta1
types.model_monitoring = types.model_monitoring_v1beta1
types.model_service = types.model_service_v1beta1
types.service_networking = types.service_networking_v1beta1
types.operation = types.operation_v1beta1
types.pipeline_failure_policy = types.pipeline_failure_policy_v1beta1
types.pipeline_job = types.pipeline_job_v1beta1
types.pipeline_service = types.pipeline_service_v1beta1
types.pipeline_state = types.pipeline_state_v1beta1
types.prediction_service = types.prediction_service_v1beta1
types.publisher_model = types.publisher_model_v1beta1
types.schedule = types.schedule_v1beta1
types.schedule_service = types.schedule_service_v1beta1
types.specialist_pool = types.specialist_pool_v1beta1
types.specialist_pool_service = types.specialist_pool_service_v1beta1
types.study = types.study_v1beta1
types.tensorboard = types.tensorboard_v1beta1
types.tensorboard_service = types.tensorboard_service_v1beta1
types.tensorboard_data = types.tensorboard_data_v1beta1
types.tensorboard_experiment = types.tensorboard_experiment_v1beta1
types.tensorboard_run = types.tensorboard_run_v1beta1
types.tensorboard_service = types.tensorboard_service_v1beta1
types.tensorboard_time_series = types.tensorboard_time_series_v1beta1
types.training_pipeline = types.training_pipeline_v1beta1
types.types = types.types_v1beta1
types.vizier_service = types.vizier_service_v1beta1
if DEFAULT_VERSION == V1:
services.dataset_service_client = services.dataset_service_client_v1
services.endpoint_service_client = services.endpoint_service_client_v1
services.feature_online_store_admin_service_client = (
services.feature_online_store_admin_service_client_v1
)
services.feature_registry_service_client = (
services.feature_registry_service_client_v1
)
services.feature_online_store_service_client = (
services.feature_online_store_service_client_v1
)
services.featurestore_online_serving_service_client = (
services.featurestore_online_serving_service_client_v1
)
services.featurestore_service_client = services.featurestore_service_client_v1
services.job_service_client = services.job_service_client_v1
services.model_garden_service_client = services.model_garden_service_client_v1
services.model_service_client = services.model_service_client_v1
services.pipeline_service_client = services.pipeline_service_client_v1
services.prediction_service_client = services.prediction_service_client_v1
services.prediction_service_async_client = (
services.prediction_service_async_client_v1
)
services.schedule_service_client = services.schedule_service_client_v1
services.specialist_pool_service_client = services.specialist_pool_service_client_v1
services.tensorboard_service_client = services.tensorboard_service_client_v1
services.index_service_client = services.index_service_client_v1
services.index_endpoint_service_client = services.index_endpoint_service_client_v1
services.vizier_service_client = services.vizier_service_client_v1
types.accelerator_type = types.accelerator_type_v1
types.annotation = types.annotation_v1
types.annotation_spec = types.annotation_spec_v1
types.artifact = types.artifact_v1
types.batch_prediction_job = types.batch_prediction_job_v1
types.completion_stats = types.completion_stats_v1
types.context = types.context_v1
types.custom_job = types.custom_job_v1
types.data_item = types.data_item_v1
types.data_labeling_job = types.data_labeling_job_v1
types.dataset = types.dataset_v1
types.dataset_service = types.dataset_service_v1
types.deployed_model_ref = types.deployed_model_ref_v1
types.encryption_spec = types.encryption_spec_v1
types.endpoint = types.endpoint_v1
types.endpoint_service = types.endpoint_service_v1
types.entity_type = types.entity_type_v1
types.env_var = types.env_var_v1
types.event = types.event_v1
types.execution = types.execution_v1
types.explanation = types.explanation_v1
types.explanation_metadata = types.explanation_metadata_v1
types.feature = types.feature_v1
types.feature_group = types.feature_group_v1
types.feature_monitoring_stats = types.feature_monitoring_stats_v1
types.feature_online_store = types.feature_online_store_v1
types.feature_online_store_admin_service = (
types.feature_online_store_admin_service_v1
)
types.feature_registry_service = types.feature_registry_service_v1
types.feature_online_store_service = types.feature_online_store_service_v1
types.feature_selector = types.feature_selector_v1
types.feature_view = types.feature_view_v1
types.feature_view_sync = types.feature_view_sync_v1
types.featurestore = types.featurestore_v1
types.featurestore_online_service = types.featurestore_online_service_v1
types.featurestore_service = types.featurestore_service_v1
types.hyperparameter_tuning_job = types.hyperparameter_tuning_job_v1
types.index = types.index_v1
types.index_endpoint = types.index_endpoint_v1
types.index_service = types.index_service_v1
types.io = types.io_v1
types.job_service = types.job_service_v1
types.job_state = types.job_state_v1
types.lineage_subgraph = types.lineage_subgraph_v1
types.machine_resources = types.machine_resources_v1
types.manual_batch_tuning_parameters = types.manual_batch_tuning_parameters_v1
types.matching_engine_deployed_index_ref = (
types.matching_engine_deployed_index_ref_v1
)
types.matching_engine_index = types.index_v1
types.matching_engine_index_endpoint = types.index_endpoint_v1
types.metadata_service = types.metadata_service_v1
types.metadata_schema = types.metadata_schema_v1
types.metadata_store = types.metadata_store_v1
types.model = types.model_v1
types.model_evaluation = types.model_evaluation_v1
types.model_evaluation_slice = types.model_evaluation_slice_v1
types.model_deployment_monitoring_job = types.model_deployment_monitoring_job_v1
types.model_monitoring = types.model_monitoring_v1
types.model_service = types.model_service_v1
types.service_networking = types.service_networking_v1
types.operation = types.operation_v1
types.pipeline_failure_policy = types.pipeline_failure_policy_v1
types.pipeline_job = types.pipeline_job_v1
types.pipeline_service = types.pipeline_service_v1
types.pipeline_state = types.pipeline_state_v1
types.prediction_service = types.prediction_service_v1
types.publisher_model = types.publisher_model_v1
types.schedule = types.schedule_v1
types.schedule_service = types.schedule_service_v1
types.specialist_pool = types.specialist_pool_v1
types.specialist_pool_service = types.specialist_pool_service_v1
types.study = types.study_v1
types.tensorboard = types.tensorboard_v1
types.tensorboard_service = types.tensorboard_service_v1
types.tensorboard_data = types.tensorboard_data_v1
types.tensorboard_experiment = types.tensorboard_experiment_v1
types.tensorboard_run = types.tensorboard_run_v1
types.tensorboard_service = types.tensorboard_service_v1
types.tensorboard_time_series = types.tensorboard_time_series_v1
types.training_pipeline = types.training_pipeline_v1
types.types = types.types_v1
types.vizier_service = types.vizier_service_v1
__all__ = (
DEFAULT_VERSION,
V1BETA1,
V1,
services,
types,
)