From 6b2d93834734b6789c13ef3782b1b3632f5c6133 Mon Sep 17 00:00:00 2001
From: Karl Weinmeister <11586922+kweinmeister@users.noreply.github.com>
Date: Fri, 5 Nov 2021 19:48:57 -0500
Subject: [PATCH] docs: Update sample with feedback from b/191251050 (#818)

---
 .../create_and_import_dataset_image_sample.py             | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/samples/model-builder/create_and_import_dataset_image_sample.py b/samples/model-builder/create_and_import_dataset_image_sample.py
index bab7c8a59c..f414f9e4ad 100644
--- a/samples/model-builder/create_and_import_dataset_image_sample.py
+++ b/samples/model-builder/create_and_import_dataset_image_sample.py
@@ -12,12 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+
+#  [START aiplatform_sdk_create_and_import_dataset_image_sample]
 from typing import List, Union
 
 from google.cloud import aiplatform
 
 
-#  [START aiplatform_sdk_create_and_import_dataset_image_sample]
 def create_and_import_dataset_image_sample(
     project: str,
     location: str,
@@ -25,6 +26,11 @@ def create_and_import_dataset_image_sample(
     src_uris: Union[str, List[str]],
     sync: bool = True,
 ):
+    """
+    src_uris -- a string or list of strings, e.g.
+        ["gs://bucket1/source1.jsonl", "gs://bucket7/source4.jsonl"]
+    """
+
     aiplatform.init(project=project, location=location)
 
     ds = aiplatform.ImageDataset.create(