From 252ac29a8cd861b93aceec5014b78cf56a5a8e1d Mon Sep 17 00:00:00 2001 From: Nassim Tabchiche Date: Thu, 26 Sep 2024 15:15:26 +0200 Subject: [PATCH] Type create_requirement_assessments method --- backend/core/models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/core/models.py b/backend/core/models.py index f3ade3331..79ea78385 100644 --- 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: