rename infer_test_set for greater clarity

llmtune/cli/toolkit.py

@@ -68,7 +68,7 @@ def run_one_experiment(config: Config, config_path: str) -> None:
     results_file_path = join(dir_helper.save_paths.results, "results.csv")
     if not exists(results_path) or exists(results_file_path):
         inference_runner = LoRAInference(test, test_column, config, dir_helper)
-        inference_runner.infer_all()
+        inference_runner.infer_test_set()
         RichUI.after_inference(results_path)
     else:
         RichUI.inference_found(results_path)

llmtune/inference/generics.py

@@ -7,5 +7,5 @@ def infer_one(self, prompt: str):
         pass
 
     @abstractmethod
-    def infer_all(self):
+    def infer_test_set(self):
         pass

llmtune/inference/lora.py

@@ -64,7 +64,7 @@ def _get_merged_model(self, weights_path: str):
 
         return model, tokenizer
 
-    def infer_all(self):
+    def infer_test_set(self):
         results = []
         prompts = self.test_dataset["formatted_prompt"]
         labels = self.test_dataset[self.label_column]