fix linting

Ssukriti · Ssukriti · commit 2222178d4f6f · 2026-01-20T12:57:28.000-07:00
Signed-off-by: Sukriti-Sharma4 &lt;sukriti.sharma4@ibm.com&gt;
diff --git a/aiu_fms_testing_utils/scripts/save_cpu_data.py b/aiu_fms_testing_utils/scripts/save_cpu_data.py
@@ -97,7 +97,7 @@ def process_row(row):
             max_new_tokens,
             LogitsExtractorHook(),
             attn_algorithm="math",
-            eos_token_id=tokenizer.eos_token_id
+            eos_token_id=tokenizer.eos_token_id,
         )
     return {"id": id, "input_ids": input_ids, "validation": cpu_validation_info}
 

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def process_row(row):`
`97`	`97`	`max_new_tokens,`
`98`	`98`	`LogitsExtractorHook(),`
`99`	`99`	`attn_algorithm="math",`
`100`		`- eos_token_id=tokenizer.eos_token_id`
	`100`	`+ eos_token_id=tokenizer.eos_token_id,`
`101`	`101`	`)`
`102`	`102`	`return {"id": id, "input_ids": input_ids, "validation": cpu_validation_info}`
`103`	`103`