From b5452e3764c5f6d51eddbf2c2bdc0cb621b78f4e Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Wed, 10 Apr 2024 11:38:26 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E9=97=AE=E9=A2=98=E6=89=B9=E9=87=8F?= =?UTF-8?q?=E6=8F=92=E5=85=A5=E5=8E=BB=E9=87=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/dataset/serializers/problem_serializers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/dataset/serializers/problem_serializers.py b/apps/dataset/serializers/problem_serializers.py index 2587e9712..cc1456206 100644 --- a/apps/dataset/serializers/problem_serializers.py +++ b/apps/dataset/serializers/problem_serializers.py @@ -60,13 +60,14 @@ class ProblemSerializers(ApiMixin, serializers.Serializer): if with_valid: self.is_valid(raise_exception=True) problem_list = self.data.get('problem_list') + problem_list = list(set(problem_list)) dataset_id = self.data.get('dataset_id') exists_problem_content_list = [problem.content for problem in QuerySet(Problem).filter(dataset_id=dataset_id, content__in=problem_list)] problem_instance_list = [Problem(id=uuid.uuid1(), dataset_id=dataset_id, content=problem_content) for problem_content in - self.data.get('problem_list') if + problem_list if (not exists_problem_content_list.__contains__(problem_content) if len(exists_problem_content_list) > 0 else True)] @@ -122,7 +123,7 @@ class ProblemSerializers(ApiMixin, serializers.Serializer): self.is_valid(raise_exception=True) problem_paragraph_mapping = QuerySet(ProblemParagraphMapping).filter(dataset_id=self.data.get("dataset_id"), problem_id=self.data.get("problem_id")) - if problem_paragraph_mapping is None or len(problem_paragraph_mapping)==0: + if problem_paragraph_mapping is None or len(problem_paragraph_mapping) == 0: return [] return native_search( QuerySet(Paragraph).filter(id__in=[row.paragraph_id for row in problem_paragraph_mapping]),