fix: LEAP-1379: disable non-functional export types for MIG projects (#…

…6657) Co-authored-by: Jo Booth <[email protected]>
HumanSignal · Nov 18, 2024 · 9c12db6 · 9c12db6
1 parent 979d8b8
commit 9c12db6
Show file tree

Hide file tree

Showing 3 changed files with 110 additions and 5 deletions.
diff --git a/label_studio/tests/export.tavern.yml b/label_studio/tests/export.tavern.yml
@@ -83,6 +83,7 @@ stages:
   response:
     status_code: 200
 
+
 ---
 test_name: YOLO_export_option_polygons
 strict: true
@@ -173,3 +174,107 @@ stages:
         name: 'ASR_MANIFEST'
         disabled: true
     status_code: 200
+
+
+---
+test_name: mig_export
+strict: false
+marks:
+- usefixtures:
+  - django_live_url
+stages:
+- id: signup
+  type: ref
+- name: stage
+  request:
+    headers:
+      content-type: application/json
+    json:
+      label_config:
+        <View>
+          <Image name="image" valueList="$images"/>
+            <RectangleLabels name="labels" toName="image">
+              <Label value="Cat"/>
+              <Label value="Dog"/>
+            </RectangleLabels>
+        </View>
+    method: POST
+    url: '{django_live_url}/api/projects'
+  response:
+    save:
+      json:
+        pk: id
+    status_code: 201
+- name: stage_export
+  request:
+    method: GET
+    url: '{django_live_url}/api/projects/{pk}/export/formats'
+  response:
+    strict:
+      - json:list_any_order
+    json:
+      - title: 'JSON'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#JSON'
+        name: 'JSON'
+      - title: 'JSON-MIN'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#JSON-MIN'
+        name: 'JSON_MIN'
+      - title: 'CSV'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#CSV'
+        name: 'CSV'
+      - title: 'TSV'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#TSV'
+        name: 'TSV'
+      - title: 'COCO'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#COCO'
+        tags: ['image segmentation', 'object detection']
+        name: 'COCO'
+        disabled: true
+      - title: 'YOLO'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#YOLO'
+        tags: ['image segmentation', 'object detection']
+        name: 'YOLO'
+        disabled: true
+      - title: 'YOLOv8 OBB'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#YOLO'
+        tags: ['image segmentation', 'object detection']
+        name: 'YOLO_OBB'
+        disabled: true
+      - title: 'CONLL2003'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#CONLL2003'
+        tags: ['sequence labeling', 'text tagging', 'named entity recognition']
+        name: 'CONLL2003'
+        disabled: true
+      - title: 'Pascal VOC XML'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#Pascal-VOC-XML'
+        tags: ['image segmentation', 'object detection']
+        name: 'VOC'
+        disabled: true
+      - title: 'Brush labels to NumPy'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#Brush-labels-to-NumPy-amp-PNG'
+        tags: ['image segmentation']
+        name: 'BRUSH_TO_NUMPY'
+        disabled: true
+      - title: 'Brush labels to PNG'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#Brush-labels-to-NumPy-amp-PNG'
+        tags: ['image segmentation']
+        name: 'BRUSH_TO_PNG'
+        disabled: true
+      - title: 'ASR Manifest'
+        description: !anystr
+        link: 'https://labelstud.io/guide/export.html#ASR-MANIFEST'
+        tags: ['speech recognition']
+        name: 'ASR_MANIFEST'
+        disabled: true
+    status_code: 200
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -203,7 +203,7 @@ django-migration-linter = "^5.1.0"
 setuptools = ">=75.4.0"
 
 # Humansignal repo dependencies
-label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/d6b2cea6a7a77ef877b44e3f8e5c36f0b96247b4.zip"}
+label-studio-sdk = {url = "https://github.com/HumanSignal/label-studio-sdk/archive/505bf58261478e9b0babb53a4815ccf8c4bd2ed9.zip"}
 
 [tool.poetry.group.test.dependencies]
 pytest = "7.2.2"