v1.0.1 fixes

Author: shaun-scale

README.rst

@@ -131,7 +131,7 @@ __ https://docs.scale.com/reference#batch-creation
         name = 'batch_name_01_07_2021'
     )
 
-Finalize Batceh
+Finalize Batch
 ^^^^^^^^^^^^^^^
 
 Finalize a Batch. Check out `Scale's API documentation`__ for more information.

scaleapi/__init__.py

@@ -5,7 +5,7 @@
 from .projects import Project
 
 TASK_TYPES = [
-    'annotation',	
+    'annotation',
     'audiotranscription',
     'categorization',
     'comparison',
@@ -18,6 +18,7 @@
     'polygonannotation',
     'segmentannotation',
     'transcription',
+    'textcollection',
     'documenttranscription',
     'videoannotation',
     'videoboxannotation',
@@ -161,7 +162,7 @@ def create_batch(self, project, batch_name, callback):
         payload = dict(project=project, name=batch_name, callback=callback)
         batchdata = self._postrequest('batches', payload)
         return Batch(batchdata, self)
-    
+
     def finalize_batch(self, batch_name):
         batchdata = self._postrequest('batches/%s/finalize' % batch_name)
         return Batch(batchdata, self)
@@ -184,7 +185,8 @@ def list_batches(self, **kwargs):
         response = self._getrequest('batches', params=kwargs)
         docs = [Batch(doc, self) for doc in response['docs']]
         return Batchlist(
-            docs, response['totalDocs'], response['limit'],response['has_more'], response.get('next_token'),
+            docs, response['totalDocs'], response['limit'], response['has_more'], response.get(
+                'next_token'),
         )
 
     def create_project(self, project_name, type, params):
@@ -195,12 +197,12 @@ def create_project(self, project_name, type, params):
     def get_projet(self, project_name):
         projectdata = self._getrequest('projects/%s' % project_name)
         return Project(projectdata, self)
-    
+
     def projects(self):
         response = self._getrequest('projects')
         return response
 
-    def update_project(self,project_name,**kwargs):
+    def update_project(self, project_name, **kwargs):
         allowed_kwargs = {'patch', 'instruction'}
         for key in kwargs:
             if key not in allowed_kwargs:
@@ -210,9 +212,6 @@ def update_project(self,project_name,**kwargs):
         return projectdata
 
 
-        
-
-
 def _AddTaskTypeCreator(task_type):
     def create_task_wrapper(self, **kwargs):
         return self.create_task(task_type, **kwargs)

setup.py

@@ -28,7 +28,7 @@
 setup(
     name='scaleapi',
     packages=['scaleapi'],
-    version='1.0.0',
+    version='1.0.1',
     description='The official Python client library for Scale AI, the Data Platform for AI',
     author='Scale AI',
     author_email='[email protected]',