Skip to content

Commit 9315f1f

Browse files
removed the todo lines
1 parent 97a16d7 commit 9315f1f

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

llmtune/cli/toolkit.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import logging
2+
import os
23
from os import listdir
34
from os.path import exists, join
45

5-
import pandas as pd
66
import torch
77
import typer
88
import yaml
@@ -78,11 +78,9 @@ def run_one_experiment(config: Config, config_path: str) -> None:
7878
RichUI.before_qa()
7979
qa_path = dir_helper.save_paths.qa
8080
if not exists(qa_path) or not listdir(qa_path):
81-
# TODO: Instantiate unit test classes
81+
# Instantiate unit test classes
8282
llm_tests = config.get("qa", {}).get("llm_tests", [])
8383
tests = QaTestRegistry.create_tests_from_list(llm_tests)
84-
# TODO: Load results.csv
85-
results_df = pd.read_csv(results_file_path)
8684
test_suite = LLMTestSuite.from_csv(results_file_path, tests)
8785
test_suite.save_test_results(os.path.join(qa_path, "unit_test_results.csv"))
8886

0 commit comments

Comments
 (0)