Merge pull request #12 from unmand-systems/add-position-option-to-exfil-poll

add position option to exfil poll

Author: Joscar-needham

setup.py

@@ -8,7 +8,7 @@ def readme():
         return f.read()
 
 setuptools.setup(name='unmand',
-    version='2.1.1',
+    version='2.1.2',
     description='Helper library for consuming the Unmand API',
     url='https://github.com/unmand-systems/unmand-python',
     author='Josiah Khor',

src/unmand/__init__.py

@@ -128,7 +128,7 @@ def queue(self, file_data, guid=None):
 
         return Extraction(None, 'FAILED')
 
-    def poll(self, extraction, max_tries=100, interval=10.0, suppress_output=False, with_probabilities=False):  # pylint: disable=too-many-branches
+    def poll(self, extraction, max_tries=100, interval=10.0, suppress_output=False, with_probabilities=False, with_positions=False):  # pylint: disable=too-many-branches
         """Check if prediction is done"""
 
         # Helper function
@@ -145,7 +145,7 @@ def estimate_extraction_time(bounding_box_count):
 
             result = requests.get(
                 f'{self.url}extractions/{extraction_guid}/data', auth=TokenAuth(self.token),
-                params={'probabilities': with_probabilities}
+                params={'probabilities': with_probabilities, 'positions': with_positions}
             )
 
             if result.status_code == requests.codes.ok:  # pylint: disable=no-member