Fix rag instance factory

openradx · Aug 5, 2024 · 6917de1 · 6917de1
1 parent 13551a4
commit 6917de1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/radis/rag/factories.py b/radis/rag/factories.py
@@ -78,18 +78,19 @@ 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
 
         self = cast(RagInstance, self)
 
         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)])