From a1fb387099dcc8a0e8e2ac6427526791b0d9063c Mon Sep 17 00:00:00 2001 From: kshitijrajsharma Date: Thu, 17 Oct 2024 12:07:04 +0200 Subject: [PATCH 1/2] Fix : Fixes status code for the training publish --- backend/core/views.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/core/views.py b/backend/core/views.py index 8c02394a..d22d09ad 100644 --- a/backend/core/views.py +++ b/backend/core/views.py @@ -729,10 +729,10 @@ def publish_training(request, training_id: int): training_instance = get_object_or_404(Training, id=training_id) if training_instance.status != "FINISHED": - return Response("Training is not FINISHED", status=404) + return Response("Training is not FINISHED", status=409) if training_instance.accuracy < 70: return Response( - "Can't publish the training since its accuracy is below 70%", status=404 + "Can't publish the training since its accuracy is below 70%", status=403 ) model_instance = get_object_or_404(Model, id=training_instance.model.id) From 558c24a6586e7a0e7700e891209735d4d8fd49b8 Mon Sep 17 00:00:00 2001 From: kshitijrajsharma Date: Thu, 17 Oct 2024 12:35:28 +0200 Subject: [PATCH 2/2] Add username in training endpoint --- backend/core/views.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/core/views.py b/backend/core/views.py index d22d09ad..c8d9c168 100644 --- a/backend/core/views.py +++ b/backend/core/views.py @@ -95,7 +95,7 @@ class DatasetViewSet( class TrainingSerializer( serializers.ModelSerializer ): # serializers are used to translate models objects to api - + user = UserSerializer(read_only=True) multimasks = serializers.BooleanField(required=False, default=False) input_contact_spacing = serializers.IntegerField( required=False, default=8, min_value=0, max_value=20