From 925bca674c05b004e291ccd3e9a2ddabfae307a3 Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Mon, 27 May 2024 14:30:37 +0200
Subject: [PATCH] Fix stale read on CreateDatasetElement

---
 arkindex/training/serializers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/arkindex/training/serializers.py b/arkindex/training/serializers.py
index b7bd7f9110..2f658f8b73 100644
--- a/arkindex/training/serializers.py
+++ b/arkindex/training/serializers.py
@@ -717,7 +717,7 @@ class DatasetElementSerializer(serializers.ModelSerializer):
         super().__init__(*args, **kwargs)
         if dataset := self.context.get("dataset"):
             self.fields["element_id"].queryset = Element.objects.filter(corpus=dataset.corpus)
-            self.fields["set"].queryset = dataset.sets.all()
+            self.fields["set"].queryset = dataset.sets.using("default")
 
     def validate_element_id(self, element):
         dataset = self.context.get("dataset")
-- 
GitLab