Skip to content

Commit 760887b

Browse files
authored
fix: Fixed docstrings for Dataset in AutoMLForecastingTrainingJob
1 parent c97199d commit 760887b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

google/cloud/aiplatform/training_jobs.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3930,7 +3930,7 @@ def run(
39303930
Supported only for tabular Datasets.
39313931
39323932
Args:
3933-
dataset (datasets.Dataset):
3933+
dataset (datasets.TimeSeriesDataset):
39343934
Required. The dataset within the same Project from which data will be used to train the Model. The
39353935
Dataset must use schema compatible with Model being trained,
39363936
and what is compatible should be described in the used
@@ -4136,7 +4136,7 @@ def _run_with_experiments(
41364136
10% for validation, and 10% for test.
41374137
41384138
Args:
4139-
dataset (datasets.Dataset):
4139+
dataset (datasets.TimeSeriesDataset):
41404140
Required. The dataset within the same Project from which data will be used to train the Model. The
41414141
Dataset must use schema compatible with Model being trained,
41424142
and what is compatible should be described in the used
@@ -4352,7 +4352,7 @@ def _run(
43524352
Supported only for tabular Datasets.
43534353
43544354
Args:
4355-
dataset (datasets.Dataset):
4355+
dataset (datasets.TimeSeriesDataset):
43564356
Required. The dataset within the same Project from which data will be used to train the Model. The
43574357
Dataset must use schema compatible with Model being trained,
43584358
and what is compatible should be described in the used

0 commit comments

Comments
 (0)