diff --git a/beat/web/algorithms/serializers.py b/beat/web/algorithms/serializers.py index bb7cb3f3fdf941eb9e2a600e31c5f1e8f9f79f0d..d44166586e4515fa8a850bd9b30865d80ae78b08 100755 --- a/beat/web/algorithms/serializers.py +++ b/beat/web/algorithms/serializers.py @@ -32,7 +32,7 @@ import beat.core.algorithm from rest_framework import serializers from operator import itemgetter -from ..code.serializers import CodeSerializer, CodeCreationSerializer +from ..code.serializers import CodeSerializer, CodeCreationSerializer, CodeModSerializer from ..libraries.serializers import LibraryReferenceSerializer from ..dataformats.serializers import ReferencedDataFormatSerializer from ..attestations.serializers import AttestationSerializer @@ -52,6 +52,12 @@ class AlgorithmCreationSerializer(CodeCreationSerializer): beat_core_class = beat.core.algorithm.Algorithm +class AlgorithmModSerializer(CodeModSerializer): + class Meta(CodeModSerializer.Meta): + model = Algorithm + beat_core_class = beat.core.algorithm.Algorithm + + # ----------------------------------------------------------