Coverage for backend/core/auxiliary/models/MLModel.py: 100%

1from django.db import models 

2from flowsheetInternals.unitops.models import SimulationObject 

3 

4from core.managers import AccessControlManager 

5class MLModel(models.Model): 

6 flowsheet = models.ForeignKey("Flowsheet", on_delete=models.CASCADE, related_name="MLModels") 

7 simulationObject = models.ForeignKey(SimulationObject, on_delete=models.CASCADE, related_name="MLModels", null=True) 

8 surrogate_model = models.JSONField(default=dict, blank=True, null=True) 

9 csv_data = models.JSONField(default=dict, blank=True, null=True) 

10 progress = models.IntegerField(default=0) 

11 charts = models.JSONField(default=list, blank=True, null=True) 

12 metrics = models.JSONField(default=list, blank=True, null=True) 

13 test_inputs = models.JSONField(default=list, blank=True, null=True) 

14 test_outputs = models.JSONField(default=list, blank=True, null=True) 

15 

16 

17 created_at = models.DateTimeField(auto_now_add=True) 

18 

19 objects = AccessControlManager() 

20 

21