Update tuning_with_checkpoints_create.py (#13562)

lakeyk · web-flow · commit 2dc04e0bb2ba · 2025-09-11T12:20:07.000-05:00
* Update tuning_with_checkpoints_create.py

* Update tuning_job_create.py
diff --git a/genai/tuning/tuning_job_create.py b/genai/tuning/tuning_job_create.py
@@ -36,7 +36,7 @@ def create_tuning_job(output_gcs_uri: str) -> str:
         metrics=[
             Metric(
                 name="FLUENCY",
-                prompt_template="""Evaluate this {response}"""
+                prompt_template="""Evaluate this {prediction}"""
             )
         ],
         output_config=OutputConfig(
diff --git a/genai/tuning/tuning_with_checkpoints_create.py b/genai/tuning/tuning_with_checkpoints_create.py
@@ -36,7 +36,7 @@ def create_with_checkpoints(output_gcs_uri: str) -> str:
         metrics=[
             Metric(
                 name="FLUENCY",
-                prompt_template="""Evaluate this {response}"""
+                prompt_template="""Evaluate this {prediction}"""
             )
         ],
         output_config=OutputConfig(

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ def create_tuning_job(output_gcs_uri: str) -> str:`
`36`	`36`	`metrics=[`
`37`	`37`	`Metric(`
`38`	`38`	`name="FLUENCY",`
`39`		`- prompt_template="""Evaluate this {response}"""`
	`39`	`+ prompt_template="""Evaluate this {prediction}"""`
`40`	`40`	`)`
`41`	`41`	`],`
`42`	`42`	`output_config=OutputConfig(`
Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ def create_with_checkpoints(output_gcs_uri: str) -> str:`
`36`	`36`	`metrics=[`
`37`	`37`	`Metric(`
`38`	`38`	`name="FLUENCY",`
`39`		`- prompt_template="""Evaluate this {response}"""`
	`39`	`+ prompt_template="""Evaluate this {prediction}"""`
`40`	`40`	`)`
`41`	`41`	`],`
`42`	`42`	`output_config=OutputConfig(`