addressing feedbacks

googleapis · Jul 16, 2021 · c139c1c · c139c1c
1 parent 59232fb
commit c139c1c
Show file tree

Hide file tree

Showing 4 changed files with 130 additions and 256 deletions.
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -36,10 +36,7 @@
     CustomJob,
     HyperparameterTuningJob,
 )
-from google.cloud.aiplatform.tensorboard import (
-    Tensorboard,
-    TensorBoardUploader,
-)
+from google.cloud.aiplatform.tensorboard import Tensorboard
 from google.cloud.aiplatform.training_jobs import (
     CustomTrainingJob,
     CustomContainerTrainingJob,
@@ -96,5 +93,4 @@
     "TimeSeriesDataset",
     "VideoDataset",
     "Tensorboard",
-    "TensorBoardUploader",
 )
diff --git a/google/cloud/aiplatform/tensorboard/__init__.py b/google/cloud/aiplatform/tensorboard/__init__.py
@@ -16,10 +16,6 @@
 #
 
 from google.cloud.aiplatform.tensorboard.tensorboard import Tensorboard
-from google.cloud.aiplatform.tensorboard.uploader import TensorBoardUploader
 
 
-__all__ = (
-    "Tensorboard",
-    "TensorBoardUploader",
-)
+__all__ = ("Tensorboard",)