Coverage for backend/core/auxiliary/serializers/MLModelSerializer.py: 56%

1from flowsheetInternals.graphicData.serializers.groupingSerializer import GroupingSerializer 

2from rest_framework import serializers 

3from core.auxiliary.models.MLModel import MLModel 

4 

5class MLModelSerializer(serializers.ModelSerializer): 

6 surrogate_model = serializers.JSONField(required=False) 

7 class Meta: 

8 model = MLModel 

9 fields = '__all__' 

10 

11 def create(self, validated_data): 

12 instance = super().create(validated_data) 

13 if instance: 

14 instance.progress = 1 # uploaded csv successfully 

15 instance.save() 

16 return instance