From 96dfc0ed37b75b22d991e1fb58beecfa2afa4ebe Mon Sep 17 00:00:00 2001
From: Samuel Gaist <samuel.gaist@idiap.ch>
Date: Mon, 20 Apr 2020 15:22:38 +0200
Subject: [PATCH] [plotters][serializers] Set beat_core_class correctly

---
 beat/web/plotters/serializers.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/beat/web/plotters/serializers.py b/beat/web/plotters/serializers.py
index dcfe8a4a2..d7f2b0b3d 100644
--- a/beat/web/plotters/serializers.py
+++ b/beat/web/plotters/serializers.py
@@ -39,6 +39,8 @@ from ..libraries.serializers import LibraryReferenceSerializer
 from django.utils.encoding import smart_text
 
 import beat.core.plotter
+import beat.core.plotterparameter
+
 import simplejson as json
 
 
@@ -95,7 +97,7 @@ class DefaultPlotterSerializer(DynamicFieldsSerializer):
 class PlotterCreationSerializer(CodeCreationSerializer):
     class Meta(CodeCreationSerializer.Meta):
         model = Plotter
-        beat_core_class = beat.core.plotter
+        beat_core_class = beat.core.plotter.Plotter
 
 
 # ----------------------------------------------------------
@@ -146,7 +148,7 @@ class PlotterParameterCreationSerializer(ContributionCreationSerializer):
             "description",
             "fork_of",
         ]
-        # beat_core_class = beat.core.PlotterParameter
+        beat_core_class = beat.core.plotterparameter.Plotterparameter
 
     def create(self, validated_data):
         plotterparameter = None
-- 
GitLab