Merge pull request #1 from nathanielherman/master

Add polygon and line annotation support to python client

Author: alexandrwang

scaleapi/__init__.py

@@ -14,6 +14,10 @@
                                  'fields', 'choices'},
                   'annotation': {'attachment', 'attachment_type', 'instruction',
                                  'objects_to_annotate', 'with_labels', 'examples'},
+                  'polygonannotation': {'attachment', 'attachment_type', 'instruction',
+                                 'objects_to_annotate', 'with_labels', 'examples'},
+                  'lineannotation': {'attachment', 'attachment_type', 'instruction',
+                                 'objects_to_annotate', 'with_labels', 'examples'},
                   'datacollection': {'attachment', 'attachment_type', 'fields'},
                   'audiotranscription': {'attachment', 'attachment_type', 'verbatim'}}
 SCALE_ENDPOINT = 'https://api.scaleapi.com/v1/'
@@ -146,6 +150,16 @@ def create_annotation_task(self, **kwargs):
         taskdata = self._postrequest('task/annotation', payload=kwargs)
         return Task(taskdata, self)
 
+    def create_polygonannotation_task(self, **kwargs):
+        validate_payload('polygonannotation', kwargs)
+        taskdata = self._postrequest('task/polygonannotation', payload=kwargs)
+        return Task(taskdata, self)
+
+    def create_lineannotation_task(self, **kwargs):
+        validate_payload('lineannotation', kwargs)
+        taskdata = self._postrequest('task/lineannotation', payload=kwargs)
+        return Task(taskdata, self)
+
     def create_datacollection_task(self, **kwargs):
         validate_payload('datacollection', kwargs)
         taskdata = self._postrequest('task/datacollection', payload=kwargs)

tests/test_client.py

@@ -118,6 +118,42 @@ def test_annotation_fail():
             attachment_type='image')
 
 
+def test_polygonannotation_ok():
+    task = client.create_polygonannotation_task(
+        callback_url='http://www.example.com/callback',
+        instruction='Draw a tight shape around the big cow',
+        attachment_type='image',
+        attachment='http://i.imgur.com/v4cBreD.jpg',
+        objects_to_annotate=['big cow'],
+        with_labels=True)
+
+
+def test_polygonannotation_fail():
+    with pytest.raises(scaleapi.ScaleInvalidRequest):
+        client.create_polygonannotation_task(
+            callback_url='http://www.example.com/callback',
+            instruction='Draw a tight shape around the big cow',
+            attachment_type='image')
+
+
+def test_lineannotation_ok():
+    task = client.create_lineannotation_task(
+        callback_url='http://www.example.com/callback',
+        instruction='Draw a tight shape around the big cow',
+        attachment_type='image',
+        attachment='http://i.imgur.com/v4cBreD.jpg',
+        objects_to_annotate=['big cow'],
+        with_labels=True)
+
+
+def test_lineannotation_fail():
+    with pytest.raises(scaleapi.ScaleInvalidRequest):
+        client.create_lineannotation_task(
+            callback_url='http://www.example.com/callback',
+            instruction='Draw a tight shape around the big cow',
+            attachment_type='image')
+
+
 def test_datacollection_ok():
     task = client.create_datacollection_task(
         callback_url='http://www.example.com/callback',