Coverage for backend/django/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_file_name = models.CharField(max_length=255, blank=True, default="") 

10 csv_bucket = models.CharField(max_length=255, blank=True, default="") 

11 csv_object_key = models.CharField(max_length=1024, blank=True, default="") 

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

13 csv_delimiter = models.CharField(max_length=1, blank=True, default="") 

14 csv_upload_session = models.ForeignKey( 

15 "UploadSession", 

16 on_delete=models.SET_NULL, 

17 related_name="ml_models", 

18 null=True, 

19 blank=True, 

20 ) 

21 progress = models.IntegerField(default=0) 

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

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

24 test_results_bucket = models.CharField(max_length=255, blank=True, default="") 

25 test_results_key = models.CharField(max_length=1024, blank=True, default="") 

26 

27 

28 created_at = models.DateTimeField(auto_now_add=True) 

29 

30 objects = AccessControlManager() 

31 

32