Refactored code

cvat-ai · May 16, 2024 · 7e2ec85 · 7e2ec85
1 parent 79e29bb
commit 7e2ec85
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/cvat/apps/engine/serializers.py b/cvat/apps/engine/serializers.py
@@ -734,9 +734,12 @@ def create(self, validated_data):
         return job
 
     def update(self, instance, validated_data):
-        state = validated_data.get('state', instance.state)
-        stage = validated_data.get('stage', instance.stage)
-        if 'stage' in validated_data or 'state' in validated_data:
+        state = validated_data.get('state')
+        stage = validated_data.get('stage')
+        if stage or state:
+            stage = stage or instance.stage
+            state = state or instance.state
+
             if stage == models.StageChoice.ANNOTATION:
                 status = models.StatusChoice.ANNOTATION
             elif stage == models.StageChoice.ACCEPTANCE and state == models.StateChoice.COMPLETED: