Add "starred" sampling method

RolnickLab · Nov 23, 2023 · 398a20a · 398a20a
1 parent 18895fc
commit 398a20a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/ami/main/api/serializers.py b/ami/main/api/serializers.py
@@ -736,6 +736,7 @@ class Meta:
             "id",
             "name",
             "details",
+            "method",
         ]
 
 

diff --git a/ami/main/api/views.py b/ami/main/api/views.py
@@ -277,6 +277,7 @@ def star(self, _request, pk=None) -> Response:
         collection, _created = SourceImageCollection.objects.get_or_create(
             project=project,
             name="Starred Images",  # @TODO i18n and store this in constants
+            defaults={"method": "starred"},
         )
         collection.images.add(source_image)
         return Response({"collection": collection.pk, "total_images": collection.images.count()})

diff --git a/ami/main/models.py b/ami/main/models.py
@@ -1903,6 +1903,7 @@ def html(self) -> str:
     "stratified_random",
     "interval",
     "manual",
+    "starred",
     "random_from_each_event",
     "last_and_random_from_each_event",
     "greatest_file_size_from_each_event",
-Original file line number
+Diff line change
@@ Expand Up / @@ -736,6 +736,7 @@ class Meta: @@
                 "id",
                 "name",
                 "details",
+                "method",
             ]
@@ Expand Down @@