From 0befde36bfd4ff1b5161b7ceb3bb55f6e7d8ea37 Mon Sep 17 00:00:00 2001
From: Yu-Han Liu <yuhanliu@google.com>
Date: Fri, 12 Feb 2021 12:18:17 -0800
Subject: [PATCH] fix: correct text sentiment analysis sample (#222)

---
 .../create_training_pipeline_text_sentiment_analysis_sample.py  | 2 +-
 .../create_training_pipeline_text_sentiment_analysis_sample.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py b/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py
index db597c6611..744f463361 100644
--- a/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py
+++ b/.sample_configs/param_handlers/create_training_pipeline_text_sentiment_analysis_sample.py
@@ -18,7 +18,7 @@ def make_parent(parent: str) -> str:
 
 def make_training_pipeline(display_name: str, dataset_id: str, model_display_name: str) -> google.cloud.aiplatform_v1alpha1.types.training_pipeline.TrainingPipeline:
     # Use sentiment_max of 4
-    training_task_inputs_dict = {"sentiment_max": 4}
+    training_task_inputs_dict = {"sentiment_max": 10}
     training_task_inputs = to_protobuf_value(training_task_inputs_dict)
     
     training_pipeline = {
diff --git a/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py b/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py
index 9386167ab2..59e878dd9d 100644
--- a/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py
+++ b/samples/snippets/create_training_pipeline_text_sentiment_analysis_sample.py
@@ -32,7 +32,7 @@ def create_training_pipeline_text_sentiment_analysis_sample(
     client = aiplatform.gapic.PipelineServiceClient(client_options=client_options)
     # Use sentiment_max of 4
     training_task_inputs = trainingjob.definition.AutoMlTextSentimentInputs(
-        sentiment_max=4,
+        sentiment_max=10,
     ).to_value()
 
     training_pipeline = {