From 718c497e701b7dcc9077b92ab5c4871ca4cec9e4 Mon Sep 17 00:00:00 2001
From: Karl Weinmeister
Date: Wed, 3 Nov 2021 20:51:21 -0500
Subject: [PATCH] docs: Update sample with feedback from b/191251050
---
.../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(