From ac6d3944f980567d53702f0f3f5af895e29ffe4f Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Thu, 1 Aug 2024 10:08:31 +0200
Subject: [PATCH] Remove Task.image_artifact

---
 .../0010_remove_task_image_artifact.py          | 17 +++++++++++++++++
 arkindex/ponos/models.py                        |  7 -------
 arkindex/process/builder.py                     |  1 -
 3 files changed, 17 insertions(+), 8 deletions(-)
 create mode 100644 arkindex/ponos/migrations/0010_remove_task_image_artifact.py

diff --git a/arkindex/ponos/migrations/0010_remove_task_image_artifact.py b/arkindex/ponos/migrations/0010_remove_task_image_artifact.py
new file mode 100644
index 0000000000..5df2b6acbf
--- /dev/null
+++ b/arkindex/ponos/migrations/0010_remove_task_image_artifact.py
@@ -0,0 +1,17 @@
+# Generated by Django 5.0.6 on 2024-08-01 07:23
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("ponos", "0009_task_original_task"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="task",
+            name="image_artifact",
+        ),
+    ]
diff --git a/arkindex/ponos/models.py b/arkindex/ponos/models.py
index 53cdeaee98..44a0dfd638 100644
--- a/arkindex/ponos/models.py
+++ b/arkindex/ponos/models.py
@@ -287,13 +287,6 @@ class Task(models.Model):
     shm_size = models.CharField(max_length=80, blank=True, null=True, editable=False)
     command = models.TextField(blank=True, null=True)
     env = HStoreField(default=dict)
-    image_artifact = models.ForeignKey(
-        "ponos.Artifact",
-        related_name="tasks_using_image",
-        on_delete=models.SET_NULL,
-        blank=True,
-        null=True,
-    )
 
     agent = models.ForeignKey(
         Agent,
diff --git a/arkindex/process/builder.py b/arkindex/process/builder.py
index e38efd0699..ee650b0e67 100644
--- a/arkindex/process/builder.py
+++ b/arkindex/process/builder.py
@@ -104,7 +104,6 @@ class ProcessBuilder:
                 requires_gpu=requires_gpu,
                 shm_size=shm_size,
                 extra_files=extra_files,
-                image_artifact_id=artifact,
                 worker_run=worker_run,
             )
         )
-- 
GitLab