From 140f59bef6eac455f3094c9ef782ff782f3831b2 Mon Sep 17 00:00:00 2001
From: Valentin Rigal <rigal@teklia.com>
Date: Wed, 3 Apr 2024 13:42:06 +0200
Subject: [PATCH] Suggestions

---
 .../training/migrations/0008_dataset_unique_elements.py     | 6 +++---
 arkindex/training/serializers.py                            | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/arkindex/training/migrations/0008_dataset_unique_elements.py b/arkindex/training/migrations/0008_dataset_unique_elements.py
index 44224cfde6..229afdda54 100644
--- a/arkindex/training/migrations/0008_dataset_unique_elements.py
+++ b/arkindex/training/migrations/0008_dataset_unique_elements.py
@@ -11,9 +11,9 @@ def update_unique_elements(apps, schema_editor):
         models.Exists(
             DatasetElement.objects
             .filter(set__dataset_id=models.OuterRef("pk"))
-            .values("element")
-            .annotate(dups=models.Count("element"))
-            .filter(dups__gte=2)
+            .values("element_id")
+            .annotate(dupes=models.Count("element_id"))
+            .filter(dupes__gte=2)
         )
     ).update(unique_elements=False)
 
diff --git a/arkindex/training/serializers.py b/arkindex/training/serializers.py
index 9686bbb124..5faa79fc61 100644
--- a/arkindex/training/serializers.py
+++ b/arkindex/training/serializers.py
@@ -583,9 +583,9 @@ class DatasetSerializer(serializers.ModelSerializer):
         if unique is True and self.instance and (
             DatasetElement.objects
             .filter(set__dataset_id=self.instance.pk)
-            .values("element")
-            .annotate(dups=Count("element"))
-            .filter(dups__gte=2)
+            .values("element_id")
+            .annotate(dupes=Count("element_id"))
+            .filter(dupes__gte=2)
             .exists()
         ):
             raise ValidationError("Elements are currently contained by multiple sets.")
-- 
GitLab