From 6917de18c70a0601b9bb39c49968352e4c81e88d Mon Sep 17 00:00:00 2001 From: Kai Schlamp Date: Mon, 5 Aug 2024 22:10:43 +0000 Subject: [PATCH] Fix rag instance factory --- radis/rag/factories.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/radis/rag/factories.py b/radis/rag/factories.py index deef3aed..9b06c48a 100644 --- a/radis/rag/factories.py +++ b/radis/rag/factories.py @@ -78,9 +78,10 @@ class Meta: model = RagInstance task = factory.SubFactory("radis.rag.factories.RagTaskFactory") + report = factory.SubFactory("radis.reports.factories.ReportFactory") @factory.post_generation - def reports(self, create, extracted, **kwargs): + def other_reports(self, create, extracted, **kwargs): if not create: return @@ -88,8 +89,8 @@ def reports(self, create, extracted, **kwargs): if extracted: for report in extracted: - self.reports.add(report) + self.other_reports.add(report) else: from radis.reports.factories import ReportFactory - self.reports.add(*[ReportFactory() for _ in range(3)]) + self.other_reports.add(*[ReportFactory() for _ in range(3)])