From 6b3dfe4e5f79ea8088caf0c2898cab2d3cca3d17 Mon Sep 17 00:00:00 2001
From: Samuel Gaist <samuel.gaist@idiap.ch>
Date: Wed, 22 Apr 2020 11:53:43 +0200
Subject: [PATCH] [algorithms][serializers] Implement mod serializer

---
 beat/web/algorithms/serializers.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/beat/web/algorithms/serializers.py b/beat/web/algorithms/serializers.py
index bb7cb3f3f..d44166586 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
+
+
 # ----------------------------------------------------------
 
 
-- 
GitLab