feat: DIA-1980: Option to run only on tasks w/o predictions (#9286)

Co-authored-by: robot-ci-heartex <robot-ci-heartex@users.noreply.github.com>
Co-authored-by: matt-bernstein <matt-bernstein@users.noreply.github.com>
Co-authored-by: Matt Bernstein <matt@humansignal.com>

Author: robot-ci-heartex

label_studio/ml_models/tests/__init__.py

@@ -0,0 +1,48 @@
+import factory
+from core.utils.common import load_func
+from django.conf import settings
+from django.utils import timezone
+from ml_models.models import ModelRun
+
+
+class ModelRunFactory(factory.django.DjangoModelFactory):
+    """
+    Factory for `ml_models.models.ModelRun`.
+
+    Defaults are intentionally minimal; override fields in tests as needed.
+    """
+
+    project = factory.SubFactory(load_func(settings.PROJECT_FACTORY))
+    organization = factory.SelfAttribute('project.organization')
+    created_by = factory.SelfAttribute('project.created_by')
+    model_version = None
+
+    project_subset = ModelRun.ProjectSubset.HASGT
+    status = ModelRun.ModelRunStatus.PENDING
+
+    triggered_at = None
+    completed_at = None
+    predictions_updated_at = None
+
+    total_predictions = 0
+    total_correct_predictions = 0
+    total_tasks = 0
+
+    class Meta:
+        model = ModelRun
+
+    @factory.post_generation
+    def _default_timestamps(self, create, extracted, **kwargs):
+        """
+        Ensure timestamps are set when status implies they should exist.
+        """
+        if not create:
+            return
+
+        if self.status == ModelRun.ModelRunStatus.COMPLETED:
+            now = timezone.now()
+            if self.triggered_at is None:
+                self.triggered_at = now
+            if self.completed_at is None:
+                self.completed_at = now
+            self.save(update_fields=['triggered_at', 'completed_at'])

label_studio/ml_models/tests/factories.py

@@ -5,7 +5,7 @@
 from django.conf import settings
 from django.utils import timezone
 from faker import Faker
-from tasks.models import Annotation, AnnotationDraft, Prediction, Task, TaskLock
+from tasks.models import Annotation, AnnotationDraft, FailedPrediction, Prediction, Task, TaskLock
 
 
 class TaskFactory(factory.django.DjangoModelFactory):
@@ -78,6 +78,17 @@ class Meta:
         model = Prediction
 
 
+class FailedPredictionFactory(factory.django.DjangoModelFactory):
+    task = factory.SubFactory(TaskFactory)
+    project = factory.SelfAttribute('task.project')
+    message = factory.Faker('sentence')
+    error_type = factory.Faker('word')
+    model_version = factory.Faker('word')
+
+    class Meta:
+        model = FailedPrediction
+
+
 class TaskLockFactory(factory.django.DjangoModelFactory):
     task = factory.SubFactory(TaskFactory)
     user = factory.SubFactory(load_func(settings.USER_FACTORY))

label_studio/tasks/tests/factories.py

@@ -74,7 +74,7 @@ dependencies = [
     "tldextract (>=5.1.3)",
     "uuid-utils (>=0.11.0,<1.0.0)",
     ## HumanSignal repo dependencies :start
-    "label-studio-sdk @ https://github.com/HumanSignal/label-studio-sdk/archive/5a899681eaab91cb2a276ad7a6b2e05fee83478f.zip",
+    "label-studio-sdk @ https://github.com/HumanSignal/label-studio-sdk/archive/e06e45d34b628fb3846eab8d27b618b81bfb68b6.zip",
     ## HumanSignal repo dependencies :end
 ]