Skip to content

Commit e6823d1

Browse files
authoredOct 30, 2024··
Merge pull request #2058 from PierreMesure/testset_jsonl
Update report testset file extension
2 parents d2ae591 + 2d2f426 commit e6823d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎giskard/rag/report.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def save(self, folder_path: str):
152152
path = Path(folder_path)
153153
path.mkdir(exist_ok=True, parents=True)
154154
self.to_html(path / "report.html")
155-
self._testset.save(path / "testset.json")
155+
self._testset.save(path / "testset.jsonl")
156156

157157
report_details = {"recommendation": self._recommendation}
158158
with open(path / "report_details.json", "w", encoding="utf-8") as f:
@@ -195,7 +195,7 @@ def load(
195195
path = Path(folder_path)
196196
knowledge_base_meta = json.load(open(path / "knowledge_base_meta.json", "r"))
197197
knowledge_base_data = pd.read_json(path / "knowledge_base.jsonl", orient="records", lines=True)
198-
testset = QATestset.load(path / "testset.json")
198+
testset = QATestset.load(path / "testset.jsonl")
199199

200200
answers = json.load(open(path / "agent_answer.json", "r"))
201201
model_outputs = [AgentAnswer(**answer) for answer in answers]

0 commit comments

Comments
 (0)
Please sign in to comment.