From 49b574728d8618b435bfb29d60d9485003736269 Mon Sep 17 00:00:00 2001 From: Hakan Erol <47988814+hakan458@users.noreply.github.com> Date: Thu, 21 Mar 2024 10:19:43 -0700 Subject: [PATCH] feat: DIA-980: Stream results from Adala into LSE (#5604) --- .../migrations/0005_auto_20240319_1738.py | 23 +++++++++++++++++++ label_studio/ml_models/models.py | 4 +++- 2 files changed, 26 insertions(+), 1 deletion(-) create mode 100644 label_studio/ml_models/migrations/0005_auto_20240319_1738.py diff --git a/label_studio/ml_models/migrations/0005_auto_20240319_1738.py b/label_studio/ml_models/migrations/0005_auto_20240319_1738.py new file mode 100644 index 00000000000..b2445b8ada6 --- /dev/null +++ b/label_studio/ml_models/migrations/0005_auto_20240319_1738.py @@ -0,0 +1,23 @@ +# Generated by Django 3.2.23 on 2024-03-19 17:38 + +from django.db import migrations, models + + +class Migration(migrations.Migration): + + dependencies = [ + ('ml_models', '0004_modelrun_job_id'), + ] + + operations = [ + migrations.AddField( + model_name='modelrun', + name='predictions_updated_at', + field=models.DateTimeField(default=None, null=True, verbose_name='predictions updated at'), + ), + migrations.AlterField( + model_name='modelrun', + name='triggered_at', + field=models.DateTimeField(default=None, null=True, verbose_name='triggered at'), + ), + ] diff --git a/label_studio/ml_models/models.py b/label_studio/ml_models/models.py index 32a9a57d826..19e94443305 100644 --- a/label_studio/ml_models/models.py +++ b/label_studio/ml_models/models.py @@ -141,7 +141,9 @@ class ModelRunStatus(models.TextChoices): created_at = models.DateTimeField(_('created at'), auto_now_add=True) - triggered_at = models.DateTimeField(_('triggered at')) + triggered_at = models.DateTimeField(_('triggered at'), null=True, default=None) + + predictions_updated_at = models.DateTimeField(_('predictions updated at'), null=True, default=None) completed_at = models.DateTimeField(_('completed at'), null=True, default=None)