From 760887b196884707473896def9e8b69c9fc77423 Mon Sep 17 00:00:00 2001 From: Ivan Cheung Date: Tue, 2 Nov 2021 10:23:03 -0400 Subject: [PATCH] fix: Fixed docstrings for Dataset in AutoMLForecastingTrainingJob --- google/cloud/aiplatform/training_jobs.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index d750d29b3a..679bb277ab 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -3930,7 +3930,7 @@ def run( Supported only for tabular Datasets. Args: - dataset (datasets.Dataset): + dataset (datasets.TimeSeriesDataset): Required. The dataset within the same Project from which data will be used to train the Model. The Dataset must use schema compatible with Model being trained, and what is compatible should be described in the used @@ -4136,7 +4136,7 @@ def _run_with_experiments( 10% for validation, and 10% for test. Args: - dataset (datasets.Dataset): + dataset (datasets.TimeSeriesDataset): Required. The dataset within the same Project from which data will be used to train the Model. The Dataset must use schema compatible with Model being trained, and what is compatible should be described in the used @@ -4352,7 +4352,7 @@ def _run( Supported only for tabular Datasets. Args: - dataset (datasets.Dataset): + dataset (datasets.TimeSeriesDataset): Required. The dataset within the same Project from which data will be used to train the Model. The Dataset must use schema compatible with Model being trained, and what is compatible should be described in the used