feat: adding TB.gcp uploader to python-aiplatform (#368)

Author: yfang1

google/cloud/aiplatform/compat/__init__.py

@@ -35,6 +35,7 @@
  services.specialist_pool_service_client_v1beta1
  )
  services.metadata_service_client = services.metadata_service_client_v1beta1
+ services.tensorboard_service_client = services.tensorboard_service_client_v1beta1
 
  types.accelerator_type = types.accelerator_type_v1beta1
  types.annotation = types.annotation_v1beta1
@@ -71,6 +72,12 @@
  types.specialist_pool_service = types.specialist_pool_service_v1beta1
  types.training_pipeline = types.training_pipeline_v1beta1
  types.metadata_service = types.metadata_service_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
 
 if DEFAULT_VERSION == V1:
 

google/cloud/aiplatform/compat/services/__init__.py

@@ -39,6 +39,9 @@
 from google.cloud.aiplatform_v1beta1.services.metadata_service import (
  client as metadata_service_client_v1beta1,
 )
+from google.cloud.aiplatform_v1beta1.services.tensorboard_service import (
+ client as tensorboard_service_client_v1beta1,
+)
 
 from google.cloud.aiplatform_v1.services.dataset_service import (
  client as dataset_service_client_v1,
@@ -80,4 +83,5 @@
  prediction_service_client_v1beta1,
  specialist_pool_service_client_v1beta1,
  metadata_service_client_v1beta1,
+ tensorboard_service_client_v1beta1,
 )

google/cloud/aiplatform/compat/types/__init__.py

@@ -51,6 +51,12 @@
  specialist_pool_service as specialist_pool_service_v1beta1,
  training_pipeline as training_pipeline_v1beta1,
  metadata_service as metadata_service_v1beta1,
+ tensorboard_service as tensorboard_service_v1beta1,
+ tensorboard_data as tensorboard_data_v1beta1,
+ tensorboard_experiment as tensorboard_experiment_v1beta1,
+ tensorboard_run as tensorboard_run_v1beta1,
+ tensorboard_service as tensorboard_service_v1beta1,
+ tensorboard_time_series as tensorboard_time_series_v1beta1,
 )
 from google.cloud.aiplatform_v1.types import (
  accelerator_type as accelerator_type_v1,
@@ -157,4 +163,10 @@
  specialist_pool_service_v1beta1,
  training_pipeline_v1beta1,
  metadata_service_v1beta1,
+ tensorboard_service_v1beta1,
+ tensorboard_data_v1beta1,
+ tensorboard_experiment_v1beta1,
+ tensorboard_run_v1beta1,
+ tensorboard_service_v1beta1,
+ tensorboard_time_series_v1beta1,
 )

google/cloud/aiplatform/tensorboard/__init__.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2021 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
+#
+# https://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.
+#