diff --git a/samples/controlled_generation.py b/samples/controlled_generation.py index 78c422464..042209a72 100644 --- a/samples/controlled_generation.py +++ b/samples/controlled_generation.py @@ -139,7 +139,7 @@ class Choice(enum.Enum): response_mime_type="text/x.enum", response_schema=Choice ), ) - print(result) # "Keyboard" + print(result) # Keyboard # [END x_enum] def test_x_enum_raw(self): @@ -157,7 +157,7 @@ def test_x_enum_raw(self): }, ), ) - print(result) # "Keyboard" + print(result) # Keyboard # [END x_enum_raw]