Merge pull request #15 from scaleapi/ner_support

Add support for NER in client

Author: james-lennon

scaleapi/__init__.py

@@ -15,7 +15,8 @@
     'audiotranscription',
     'pointannotation',
     'cuboidannotation',
-    'segmentannotation'
+    'segmentannotation',
+    'namedentityrecognition'
 ]
 SCALE_ENDPOINT = 'https://api.scale.com/v1/'
 DEFAULT_LIMIT = 100

tests/test_client.py

@@ -209,6 +209,17 @@ def test_audiotranscription_fail2():
             attachment='some_non_url')
 
 
+def test_namedentityrecognition_ok():
+    return client.create_namedentityrecognition_task(
+        callback_url='http://www.example.com/callback',
+        instruction='Do the objects in these images have the same pattern?',
+        text='Example text to label with NER tool',
+        labels=[{
+            'name': 'Label_A',
+            'description': 'the first label',
+        }])
+
+
 def test_unicode_ok():
     task = client.create_categorization_task(
         callback_url='http://www.example.com/callback',