Type create_requirement_assessments method

intuitem · Sep 26, 2024 · 252ac29 · 252ac29
1 parent e7fbd80
commit 252ac29
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/core/models.py b/backend/core/models.py
@@ -2080,7 +2080,9 @@ def save(self, *args, **kwargs) -> None:
             self.scores_definition = self.framework.scores_definition
         super().save(*args, **kwargs)
 
-    def create_requirement_assessments(self, baseline: Self | None = None):
+    def create_requirement_assessments(
+        self, baseline: Self | None = None
+    ) -> list["RequirementAssessment"]:
         requirements = RequirementNode.objects.filter(framework=self.framework)
         requirement_assessments = []
         for requirement in requirements: