feat: add tensorboard resource management

googleapis · Jul 14, 2021 · d34b8ba · d34b8ba
1 parent bab5225
commit d34b8ba
Show file tree

Hide file tree

Showing 5 changed files with 711 additions and 1 deletion.
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -36,6 +36,10 @@
     CustomJob,
     HyperparameterTuningJob,
 )
+from google.cloud.aiplatform.tensorboard import (
+    Tensorboard,
+    TensorBoardUploader,
+)
 from google.cloud.aiplatform.training_jobs import (
     CustomTrainingJob,
     CustomContainerTrainingJob,
@@ -91,4 +95,6 @@
     "TextDataset",
     "TimeSeriesDataset",
     "VideoDataset",
+    "Tensorboard",
+    "TensorBoardUploader",
 )
diff --git a/google/cloud/aiplatform/compat/types/__init__.py b/google/cloud/aiplatform/compat/types/__init__.py
@@ -53,7 +53,7 @@
     study as study_v1beta1,
     training_pipeline as training_pipeline_v1beta1,
     metadata_service as metadata_service_v1beta1,
-    tensorboard_service as tensorboard_service_v1beta1,
+    tensorboard as tensorboard_v1beta1,
     tensorboard_data as tensorboard_data_v1beta1,
     tensorboard_experiment as tensorboard_experiment_v1beta1,
     tensorboard_run as tensorboard_run_v1beta1,
@@ -167,6 +167,7 @@
     specialist_pool_service_v1beta1,
     training_pipeline_v1beta1,
     metadata_service_v1beta1,
+    tensorboard_v1beta1,
     tensorboard_service_v1beta1,
     tensorboard_data_v1beta1,
     tensorboard_experiment_v1beta1,

diff --git a/google/cloud/aiplatform/tensorboard/__init__.py b/google/cloud/aiplatform/tensorboard/__init__.py
@@ -14,3 +14,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+
+from google.cloud.aiplatform.tensorboard.tensorboard import Tensorboard
+from google.cloud.aiplatform.tensorboard.uploader import TensorBoardUploader
+
+
+__all__ = (
+    "Tensorboard",
+    "TensorBoardUploader",
+)