Suggestions

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)
 

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.")