From 5fa243358ec5688a40ef0d5abf3802acfa931ea7 Mon Sep 17 00:00:00 2001
From: Franklin Nunez <69214580+b-loved-dreamer@users.noreply.github.com>
Date: Fri, 3 Sep 2021 09:13:48 -0700
Subject: [PATCH] docs(samples): fixed model adaptation sample to correctly
 refer to custom classes (#246)

* fix: fixed model adaptation sample to correctly refer to custom classes

* fix: fixed model adaptation sample to correctly refer to custom classes
---
 samples/snippets/speech_model_adaptation_beta.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/samples/snippets/speech_model_adaptation_beta.py b/samples/snippets/speech_model_adaptation_beta.py
index 49fd4ad2..a3ac4e9d 100644
--- a/samples/snippets/speech_model_adaptation_beta.py
+++ b/samples/snippets/speech_model_adaptation_beta.py
@@ -34,7 +34,7 @@ def transcribe_with_model_adaptation(
     parent = f"projects/{project_id}/locations/{location}"
 
     # Create the custom class resource
-    custom_class_response = adaptation_client.create_custom_class(
+    adaptation_client.create_custom_class(
         {
             "parent": parent,
             "custom_class_id": custom_class_id,
@@ -47,7 +47,9 @@ def transcribe_with_model_adaptation(
             },
         }
     )
-    custom_class_name = custom_class_response.name
+    custom_class_name = (
+        f"projects/{project_id}/locations/{location}/customClasses/{custom_class_id}"
+    )
     # Create the phrase set resource
     phrase_set_response = adaptation_client.create_phrase_set(
         {
@@ -55,7 +57,9 @@ def transcribe_with_model_adaptation(
             "phrase_set_id": phrase_set_id,
             "phrase_set": {
                 "boost": 10,
-                "phrases": [{"value": f"Visit restaurants like ${custom_class_name}"}],
+                "phrases": [
+                    {"value": f"Visit restaurants like ${{{custom_class_name}}}"}
+                ],
             },
         }
     )
@@ -88,4 +92,5 @@ def transcribe_with_model_adaptation(
         print("Transcript: {}".format(result.alternatives[0].transcript))
 
     # [END speech_transcribe_with_model_adaptation]
+
     return response.results[0].alternatives[0].transcript