Skip to content
Snippets Groups Projects
Commit a050d4a1 authored by Erwan Rouchet's avatar Erwan Rouchet Committed by Bastien Abadie
Browse files

Fix 4 stale reads in CreateClassification

Closes #507
parent 735b3801
No related branches found
No related tags found
1 merge request!1051Fix 4 stale reads in CreateClassification
......@@ -112,6 +112,8 @@ class ClassificationCreateSerializer(serializers.ModelSerializer):
"""
Serializer to create a single classification, defaulting to manual
"""
element = serializers.PrimaryKeyRelatedField(queryset=Element.objects.using('default'))
ml_class = serializers.PrimaryKeyRelatedField(queryset=MLClass.objects.using('default'))
worker_version = serializers.PrimaryKeyRelatedField(queryset=WorkerVersion.objects.all(), default=None)
confidence = serializers.FloatField(
min_value=0,
......@@ -143,11 +145,11 @@ class ClassificationCreateSerializer(serializers.ModelSerializer):
read_only_fields = ('id', 'state')
validators = [
UniqueTogetherValidator(
queryset=Classification.objects.filter(worker_version__isnull=False, source_id__isnull=True),
queryset=Classification.objects.using('default').filter(worker_version__isnull=False, source_id__isnull=True),
fields=['element', 'worker_version', 'ml_class']
),
UniqueTogetherValidator(
queryset=Classification.objects.filter(worker_version__isnull=True, source_id__isnull=True),
queryset=Classification.objects.using('default').filter(worker_version__isnull=True, source_id__isnull=True),
fields=['element', 'ml_class']
)
]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment