Skip to content

Commit

Permalink
Merge pull request #2058 from PierreMesure/testset_jsonl
Browse files Browse the repository at this point in the history
Update report testset file extension
  • Loading branch information
henchaves authored Oct 30, 2024
2 parents d2ae591 + 2d2f426 commit e6823d1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions giskard/rag/report.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ def save(self, folder_path: str):
path = Path(folder_path)
path.mkdir(exist_ok=True, parents=True)
self.to_html(path / "report.html")
self._testset.save(path / "testset.json")
self._testset.save(path / "testset.jsonl")

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

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

0 comments on commit e6823d1

Please sign in to comment.