From c0543cdd1e9ba0efd18d7d1a442906938fc6db9a Mon Sep 17 00:00:00 2001
From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
Date: Fri, 11 Jun 2021 14:48:09 -0400
Subject: [PATCH] fix: add target_column docstring (#473)

---
 google/cloud/aiplatform/training_jobs.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py
index 57329fe937..cf33b23ba4 100644
--- a/google/cloud/aiplatform/training_jobs.py
+++ b/google/cloud/aiplatform/training_jobs.py
@@ -2583,6 +2583,8 @@ def run(
                 [google.cloud.aiplatform.v1beta1.TrainingPipeline.training_task_definition].
                 For tabular Datasets, all their data is exported to
                 training, to pick and choose from.
+            target_column (str):
+                Required. The name of the column values of which the Model is to predict.
             training_fraction_split (float):
                 Required. The fraction of the input data that is to be
                 used to train the Model. This is ignored if Dataset is not provided.
@@ -2698,6 +2700,8 @@ def _run(
                 [google.cloud.aiplatform.v1beta1.TrainingPipeline.training_task_definition].
                 For tabular Datasets, all their data is exported to
                 training, to pick and choose from.
+            target_column (str):
+                Required. The name of the column values of which the Model is to predict.
             training_fraction_split (float):
                 Required. The fraction of the input data that is to be
                 used to train the Model. This is ignored if Dataset is not provided.