Allow setting any readable model version as parent of a model version, not just versions from the same model

closes #1708 (closed)

arkindex/training/api.py

@@ -175,7 +175,6 @@ class ModelVersionsList(ListCreateAPIView):
         context.update({
             "model": self.model,
             "is_contributor": self.access_level and self.access_level >= Role.Contributor.value,
-            "user": self.request.user
         })
         return context
 
@@ -216,7 +215,6 @@ class ModelVersionsRetrieve(RetrieveUpdateDestroyAPIView):
         access_level = getattr(self, "access_level", None)
         context.update({
             "is_contributor": access_level is not None and access_level >= Role.Contributor.value,
-            "user": self.request.user
         })
         return context
 
@@ -309,10 +307,15 @@ class ValidateModelVersion(TrainingModelMixin, GenericAPIView):
             # Set the current model version as erroneous and return the available one
             instance.state = ModelVersionState.Error
             instance.save(update_fields=["state"])
+            # Set context
+            context = {
+                **self.get_serializer_context(),
+                "is_contributor": True
+            }
             return Response(
                 ModelVersionSerializer(
                     existing_model_version,
-                    context={"is_contributor": True, "model": instance, "user": request.user},
+                    context=context,
                 ).data,
                 status=status.HTTP_409_CONFLICT,
             )