From b661667b58d08ba9088b06ea9882340a0b25ed4c Mon Sep 17 00:00:00 2001
From: Max Tian <max.tian@servicenow.com>
Date: Thu, 16 Nov 2023 09:30:38 -0500
Subject: [PATCH] better naming convention

---
 bigcode_eval/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bigcode_eval/utils.py b/bigcode_eval/utils.py
index 394aa44f5..995044202 100644
--- a/bigcode_eval/utils.py
+++ b/bigcode_eval/utils.py
@@ -234,7 +234,7 @@ def complete_code(
     instruction_tokens=None,
     postprocess=True,
     is_wrapped=False,
-    save_every_k_samples: int = -1,
+    save_every_k_tasks: int = -1,
     intermediate_save_generations_path: Optional[str] = None,
     **gen_kwargs,
 ):
@@ -315,7 +315,7 @@ def complete_code(
             for sample, generated_tokens in zip(generated_tasks, generated_tokens):
                 gen_token_dict[sample].append(generated_tokens)
 
-            if save_every_k_samples >= 1 and (step + 1) % save_every_k_samples == 0:
+            if save_every_k_tasks >= 1 and (step + 1) % save_every_k_tasks == 0:
                 if not intermediate_save_generations_path:
                     raise ValueError(
                         "intermediate_save_generations_path cannot be empty!"